diff --git a/Cargo.lock b/Cargo.lock
index da9a3565fd2d..5785ec510dc3 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2152,6 +2152,7 @@ name = "wasmtime-jit"
 version = "0.9.0"
 dependencies = [
  "anyhow",
+ "cfg-if",
  "cranelift-codegen",
  "cranelift-entity",
  "cranelift-frontend",
diff --git a/crates/jit/Cargo.toml b/crates/jit/Cargo.toml
index a4a1c59a4cd1..f93b1d3173dc 100644
--- a/crates/jit/Cargo.toml
+++ b/crates/jit/Cargo.toml
@@ -25,6 +25,7 @@ target-lexicon = { version = "0.9.0", default-features = false }
 wasmparser = { version = "0.45.1", default-features = false }
 more-asserts = "0.2.1"
 anyhow = "1.0"
+cfg-if = "0.1.9"
 
 [target.'cfg(target_os = "windows")'.dependencies]
 winapi = { version = "0.3.7", features = ["winnt", "impl-default"] }
diff --git a/crates/jit/src/link.rs b/crates/jit/src/link.rs
index f8c9c6258905..d73729d94011 100644
--- a/crates/jit/src/link.rs
+++ b/crates/jit/src/link.rs
@@ -335,16 +335,7 @@ fn relocate(
                         FloorF64 => wasmtime_f64_floor as usize,
                         TruncF64 => wasmtime_f64_trunc as usize,
                         NearestF64 => wasmtime_f64_nearest as usize,
-                        #[cfg(not(target_os = "windows"))]
-                        Probestack => __rust_probestack as usize,
-                        #[cfg(all(target_os = "windows", target_env = "gnu"))]
-                        Probestack => ___chkstk as usize,
-                        #[cfg(all(
-                            target_os = "windows",
-                            target_env = "msvc",
-                            target_pointer_width = "64"
-                        ))]
-                        Probestack => __chkstk as usize,
+                        Probestack => PROBESTACK as usize,
                         other => panic!("unexpected libcall: {}", other),
                     }
                 }
@@ -398,19 +389,33 @@ fn relocate(
     }
 }
 
-/// A declaration for the stack probe function in Rust's standard library, for
-/// catching callstack overflow.
-extern "C" {
-    #[cfg(not(target_os = "windows"))]
-    pub fn __rust_probestack();
-    #[cfg(all(
-        target_os = "windows",
-        target_env = "msvc",
-        target_pointer_width = "64"
-    ))]
-    pub fn __chkstk();
-    // ___chkstk (note the triple underscore) is implemented in compiler-builtins/src/x86_64.rs
-    // by the Rust compiler for the MinGW target
-    #[cfg(all(target_os = "windows", target_env = "gnu"))]
-    pub fn ___chkstk();
+// A declaration for the stack probe function in Rust's standard library, for
+// catching callstack overflow.
+cfg_if::cfg_if! {
+    if #[cfg(any(
+        target_arch="aarch64",
+        all(
+            target_os = "windows",
+            target_env = "msvc",
+            target_pointer_width = "64"
+        )
+    ))] {
+        extern "C" {
+            pub fn __chkstk();
+        }
+        const PROBESTACK: unsafe extern "C" fn() = __chkstk;
+    } else if #[cfg(all(target_os = "windows", target_env = "gnu"))] {
+        extern "C" {
+            // ___chkstk (note the triple underscore) is implemented in compiler-builtins/src/x86_64.rs
+            // by the Rust compiler for the MinGW target
+            #[cfg(all(target_os = "windows", target_env = "gnu"))]
+            pub fn ___chkstk();
+        }
+        const PROBESTACK: unsafe extern "C" fn() = ___chkstk;
+    } else {
+        extern "C" {
+            pub fn __rust_probestack();
+        }
+        static PROBESTACK: unsafe extern "C" fn() = __rust_probestack;
+    }
 }