diff --git a/winch/codegen/build.rs b/winch/codegen/build.rs index 9f18db48a0..378ac51732 100644 --- a/winch/codegen/build.rs +++ b/winch/codegen/build.rs @@ -3,10 +3,14 @@ fn main() { return; } - if cfg!(target_arch = "x86_64") { + let arch = std::env::var("CARGO_CFG_TARGET_ARCH").unwrap(); + + if arch == "x86_64" { println!("cargo:rustc-cfg=feature=\"x64\""); - } else if cfg!(target_arch = "aarch64") { + } else if arch == "aarch64" { println!("cargo:rustc-cfg=feature=\"arm64\""); + } else { + println!("cargo:rustc-cfg=feature=\"{arch}\""); } println!("cargo:rerun-if-changed=build.rs"); }