c8h4/ZLUDA
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Fix type errors.

Browse source
This commit is contained in:
Seunghoon Lee 2024-03-25 14:09:02 +09:00
parent 1122cc0e83
commit 0f191c2354
No known key found for this signature in database
GPG key ID: 91024D13C6CA4722
1 changed files with 8 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
zluda_blas/src/lib.rs
View file

@ -285,26 +285,21 @@ unsafe fn sdot(
incx: i32,
y: *const f32,
incy: i32,
) -> cublasStatus_t {
) -> f32 {
let mut handle = mem::zeroed();
let mut status = to_cuda(rocblas_create_handle(handle));
if status != cublasStatus_t::CUBLAS_STATUS_SUCCESS {
return status;
}
let result = mem::zeroed();
status = to_cuda(rocblas_sdot(
let mut status = rocblas_create_handle(handle);
let mut result = 0.0;
status = rocblas_sdot(
handle.cast(),
n,
x,
incx,
y,
incy,
result,
));
if status != cublasStatus_t::CUBLAS_STATUS_SUCCESS {
return status;
}
to_cuda(rocblas_destroy_handle(*handle))
&mut result,
);
status = rocblas_destroy_handle(*handle);
result
}
unsafe fn dasum_v2(