Put the core::arch::asm invocation under the cfg(target_os = "zkvm"…

…) for less space for the wrong signature error
openvm-org · Jan 8, 2025 · 9c55cf9 · 9c55cf9
1 parent 99b0046
commit 9c55cf9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/crates/toolchain/platform/custom_insn/src/lib.rs b/crates/toolchain/platform/custom_insn/src/lib.rs
@@ -318,6 +318,7 @@ pub fn custom_insn_r(input: proc_macro::TokenStream) -> proc_macro::TokenStream
     handle_reg_arg(&mut template, &mut args, &rs2, "rs2");
 
     let expanded = quote::quote! {
+        #[cfg(target_os = "zkvm")]
         unsafe {
             core::arch::asm!(
                 #template,
@@ -374,6 +375,7 @@ pub fn custom_insn_i(input: proc_macro::TokenStream) -> proc_macro::TokenStream
     handle_reg_arg(&mut template, &mut args, &imm, "imm");
 
     let expanded = quote::quote! {
+        #[cfg(target_os = "zkvm")]
         unsafe {
             core::arch::asm!(
                 #template,