refactor(rust): Add missing feature flag for prefetching on arm (#17740)

pola-rs · Jul 19, 2024 · 7015663 · 7015663
1 parent 690af34
commit 7015663
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/crates/polars-utils/src/lib.rs b/crates/polars-utils/src/lib.rs
@@ -1,3 +1,7 @@
+#![cfg_attr(
+    all(target_arch = "aarch64", feature = "nightly"),
+    feature(stdarch_aarch64_prefetch)
+)]
 #![cfg_attr(docsrs, feature(doc_auto_cfg))]
 pub mod abs_diff;
 pub mod arena;

diff --git a/crates/polars-utils/src/mem.rs b/crates/polars-utils/src/mem.rs
@@ -17,7 +17,7 @@ pub unsafe fn prefetch_l2(ptr: *const u8) {
         unsafe { _mm_prefetch(ptr as *const _, _MM_HINT_T1) };
     }
 
-    #[cfg(target_arch = "aarch64")]
+    #[cfg(all(target_arch = "aarch64", feature = "nightly"))]
     {
         use std::arch::aarch64::*;
         unsafe { _prefetch(ptr as *const _, _PREFETCH_READ, _PREFETCH_LOCALITY2) };