From d9f93d31b4901520ad065f329748710625450502 Mon Sep 17 00:00:00 2001
From: David Chavez <david@dcvz.io>
Date: Mon, 4 Dec 2023 19:19:46 +0100
Subject: [PATCH] test burn-ndarray also with accelerate (#1050)

---
 xtask/src/runchecks.rs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/xtask/src/runchecks.rs b/xtask/src/runchecks.rs
index 89c2249fd..3985d0b43 100644
--- a/xtask/src/runchecks.rs
+++ b/xtask/src/runchecks.rs
@@ -275,13 +275,18 @@ fn burn_dataset_features_std() {
     endgroup!();
 }
 
-// Test burn-candle with accelerate (macOS only)
-// Leverages the macOS Accelerate framework: https://developer.apple.com/documentation/accelerate
+// macOS only checks
 #[cfg(target_os = "macos")]
-fn burn_candle_accelerate() {
+fn macos_checks() {
+    // Leverages the macOS Accelerate framework: https://developer.apple.com/documentation/accelerate
     group!("Checks: burn-candle (accelerate)");
     cargo_test(["-p", "burn-candle", "--features", "accelerate"].into());
     endgroup!();
+
+    // Leverages the macOS Accelerate framework: https://developer.apple.com/documentation/accelerate
+    group!("Checks: burn-ndarray (accelerate)");
+    cargo_test(["-p", "burn-ndarray", "--features", "blas-accelerate"].into());
+    endgroup!();
 }
 
 fn std_checks() {
@@ -328,7 +333,7 @@ fn std_checks() {
 
     // Test burn-candle with accelerate (macOS only)
     #[cfg(target_os = "macos")]
-    burn_candle_accelerate();
+    macos_checks();
 
     // Test burn-dataset features
     burn_dataset_features_std();