huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename vec-dot to vec-ops. (#449)

Browse Source 
* Rename vec-dot to vec-ops.

* Also bump the crate version.

* Add a currently empty readme.
This commit is contained in:
Laurent Mazare
2023-08-15 10:48:57 +01:00
committed by GitHub
parent 495e0b7580
commit 531f23b4d0
14 changed files with 30 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
candle-core/src/cpu/kernels.rs
View File

@ -1,4 +1,4 @@
pub trait VecDot: num_traits::NumAssign + Copy {
pub trait VecOps: num_traits::NumAssign + Copy {
/// Dot-product of two vectors.
///
/// # Safety
@ -28,7 +28,7 @@ pub trait VecDot: num_traits::NumAssign + Copy {
}
}
impl VecDot for f32 {
impl VecOps for f32 {
#[inline(always)]
unsafe fn vec_dot(lhs: *const Self, rhs: *const Self, res: *mut Self, len: usize) {
super::vec_dot_f32(lhs, rhs, res, len)
@ -40,7 +40,7 @@ impl VecDot for f32 {
}
}
impl VecDot for half::f16 {
impl VecOps for half::f16 {
#[inline(always)]
unsafe fn vec_dot(lhs: *const Self, rhs: *const Self, res: *mut Self, len: usize) {
let mut res_f32 = 0f32;
@ -49,10 +49,10 @@ impl VecDot for half::f16 {
}
}
impl VecDot for f64 {}
impl VecDot for half::bf16 {}
impl VecDot for u8 {}
impl VecDot for u32 {}
impl VecOps for f64 {}
impl VecOps for half::bf16 {}
impl VecOps for u8 {}
impl VecOps for u32 {}
#[inline(always)]
pub fn par_for_each(n_threads: usize, func: impl Fn(usize) + Send + Sync) {