Skip to content

Commit 224af41

Browse files
committed
clean code
1 parent 6849bf1 commit 224af41

3 files changed

Lines changed: 94 additions & 232 deletions

File tree

csrc/pythonInterface.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,7 @@ void gemm_4bit_inference_bf16(
384384
int m, int n, int k, int l, sycl::ext::oneapi::bfloat16 * A, unsigned char* B, float *absmax, float *datatype, float * out,
385385
int lda, int ldb, int ldc, int blocksize, sycl::queue* stream
386386
) {
387-
gemm_4bit_inference_cutlass_dequant<sycl::ext::oneapi::bfloat16, 16>(m, n, k, l, A, B, absmax, datatype, out, lda, ldb, ldc, blocksize, stream);
387+
gemm_4bit_cutlass<sycl::ext::oneapi::bfloat16, 16>(m, n, k, l, A, B, absmax, datatype, out, lda, ldb, ldc, blocksize, stream);
388388
}
389389
#endif
390390

csrc/xpu_cutlass.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ void gemv_4bit_inference_cutlass_cute(int m, int n, int k, T *A, T *B,
108108
int ldb, int ldc, int blocksize, sycl::queue *stream);
109109

110110
template <typename T, int BITS>
111-
void gemm_4bit_inference_cutlass_dequant(int m, int n, int k, int l, T *A, unsigned char *B,
111+
void gemm_4bit_cutlass(int m, int n, int k, int l, T *A, unsigned char *B,
112112
float *absmax, float *datatype, float *out, int lda,
113113
int ldb, int ldc, int blocksize, sycl::queue *stream);
114114

0 commit comments

Comments
 (0)