wassname/vllm
1
0
Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-08-15 12:54:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Kernel] Squash a few more warnings (#6914)

Browse Source
This commit is contained in:
Tyler Michael Smith
2024-07-30 13:50:42 -04:00
committed by GitHub
parent 5cf9254a9c
commit cbbc904470
5 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
csrc/quantization/squeezellm/quant_cuda_kernel.cu
+2 -1
View File
@@ -203,7 +203,8 @@ void squeezellm_gemm(torch::Tensor vec, torch::Tensor mat, torch::Tensor mul,
#endif
mat.data_ptr<int>(),
#ifndef USE_ROCM
(half2*)mul.data<at::Half>(), (__half*)lookup_table.data_ptr<at::Half>(),
(half2*)mul.data_ptr<at::Half>(),
(__half*)lookup_table.data_ptr<at::Half>(),
#else
(float2*)mul.data_ptr<float>(),
(__half*)lookup_table.data_ptr<at::Half>(),