Skip to content

Commit

Permalink
a
Browse files Browse the repository at this point in the history
  • Loading branch information
eddyxu committed Sep 15, 2024
1 parent d2df667 commit 54c7009
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions rust/lance-linalg/src/distance/l2.rs
Original file line number Diff line number Diff line change
Expand Up @@ -144,9 +144,7 @@ impl L2 for f16 {
#[cfg(target_arch = "aarch64")]
{
if std::arch::is_aarch64_feature_detected!("neon") {
if std::arch::is_aarch64_feature_detected!("fp16") {
return unsafe { kernel::l2_f16_neon(x.as_ptr(), y.as_ptr(), x.len() as u32) };
}
return unsafe { kernel::l2_f16_neon(x.as_ptr(), y.as_ptr(), x.len() as u32) };
}
}

Expand Down

0 comments on commit 54c7009

Please sign in to comment.