dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test burn-ndarray also with accelerate (#1050)

This commit is contained in:
David Chavez 2023-12-04 19:19:46 +01:00 committed by GitHub
parent 82f5722ca1
commit d9f93d31b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
xtask/src/runchecks.rs
View File

@ -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();