MCPcopy Create free account

hub / github.com/Lyn-liyuan/ndarray-cuda-matmul / functions

Functions56 in github.com/Lyn-liyuan/ndarray-cuda-matmul

↓ 25 callersMethodshape
(&self)
src/lib.rs:57
↓ 16 callersMethodto_device
(&self)
src/lib.rs:294
↓ 11 callersMethodto_host
(&self)
src/lib.rs:354
↓ 10 callersMethoddot
(&self, mat: &CudaMat<Ix1>)
src/lib.rs:90
↓ 8 callersFunctiondestory_cublas
Wrapper function to destroy/free CUDA resources.
src/lib.rs:391
↓ 8 callersFunctioninit_cublas
Wrapper function to initialize CUDA for matrix operations.
src/lib.rs:384
↓ 6 callersMethodcuda_dot
Performs dot product on 1D arrays using CUDA and returns the result as a 1-element array.
src/lib.rs:242
↓ 4 callersFunctionmatmul
Function to perform matrix multiplication using the cuBLAS library.
src/lib.rs:396
↓ 3 callersMethodt
(&self)
src/lib.rs:203
↓ 2 callersFunctionget_shape
Helper function to determine the shape of the resulting matrix after multiplication.
src/lib.rs:414
↓ 2 callersMethodinv
(&self)
src/lib.rs:178
↓ 1 callersFunctiondo_cuda
(x: &Array2<f32>, y: &Array2<f32>)
examples/MultipleLinearRegression/src/main.rs:61
↓ 1 callersFunctiondo_ndarray_linalg
(x: &Array2<f32>, y: &Array2<f32>)
examples/MultipleLinearRegression/src/main.rs:43
↓ 1 callersFunctiondo_run
(a:&ArrayBase<OwnedRepr<f32>, Ix1>)
src/lib.rs:553
↓ 1 callersFunctiondot_with_cuda
()
src/lib.rs:525
↓ 1 callersFunctiondot_with_device
()
src/lib.rs:535
↓ 1 callersFunctiondot_with_ndarry
()
src/lib.rs:515
↓ 1 callersFunctiongen_data
()
examples/MultipleLinearRegression/src/main.rs:7
↓ 1 callersMethodmul_scalar
(&self, scalar: f32)
src/lib.rs:60
↓ 1 callersFunctionscatter
(matrix: &Array2<f32>,y:&Array2<f32>)
examples/pca/src/main.rs:73
↓ 1 callersFunctionshow_mat
(matrix: &Array2<f32>)
examples/pca/src/main.rs:14
Functionbenchmark_kernel
src/common.h:119
Functionceil_div
src/common.h:8
FunctioncuSolverCheck
cuSolver error checking
src/common.h:36
FunctioncublasCheck
cuBLAS error checking
src/common.h:26
Functioncuda_check
CUDA error checking
src/common.h:16
Methoddrop
(&mut self)
src/lib.rs:49
Functionix1_dot_ix1
()
src/lib.rs:665
Functionix1_dot_ix1_device
()
src/lib.rs:609
Functionix1_dot_ix1_unmatch
()
src/lib.rs:674
Functionix1_dot_ix2
()
src/lib.rs:680
Functionix1_dot_ix2_device
()
src/lib.rs:619
Functionix2_dot_ix1
()
src/lib.rs:687
Functionix2_dot_ix2
()
src/lib.rs:694
Functionix2_dot_ix2_device
()
src/lib.rs:628
Functionix2_dot_ix2_run
()
src/lib.rs:560
Functionix2_inv_device
()
src/lib.rs:649
Functionleast_squares_method
()
src/lib.rs:588
Functionmain
()
build.rs:4
Functionmain
()
examples/MultipleLinearRegression/build.rs:1
Functionmain
()
examples/MultipleLinearRegression/src/main.rs:70
Functionmain
()
examples/pca/build.rs:1
Functionmain
()
examples/pca/src/main.rs:102
Functionmake_ones_float
src/common.h:78
Functionmake_random_float
src/common.h:56
Functionmake_random_float_01
src/common.h:48
Functionmake_random_int
src/common.h:64
Functionmake_zeros_float
src/common.h:72
Functionmat_t
()
src/lib.rs:599
Functionperformance
()
src/lib.rs:707
Functionrun1
(array: &A, func: F)
src/lib.rs:442
Functionrun2
(a: &A, b: &B, func: F)
src/lib.rs:456
Functionrun3
(a: &A, b: &B, c: &C, func: F)
src/lib.rs:473
Functionsolving
()
src/lib.rs:578
Functiontest_run
()
src/lib.rs:548
Functionvalidate_result
src/common.h:90