diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-25 14:02:14 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-07-25 14:02:14 -0700 |
| commit | 8b1fd32e3e4f5073fd055cb5f9261ec585f8cc2c (patch) | |
| tree | 76044424d73d02e1026c996b22b9da5061188387 /csrc | |
| parent | 7d2ecd30c044840ba5f161ec73e5eaf30ac8131d (diff) | |
Fixed makefile; fixed Ampere igemmlt_8 bug.
Diffstat (limited to 'csrc')
| -rw-r--r-- | csrc/pythonInterface.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/csrc/pythonInterface.c b/csrc/pythonInterface.c index 03c8d92..9b57549 100644 --- a/csrc/pythonInterface.c +++ b/csrc/pythonInterface.c @@ -228,7 +228,7 @@ extern "C" { return igemmlt_ampere_8_rowscale((cublasLtHandle_t) context->m_handle, m, n, k, A, B, C, row_scale, lda, ldb, ldc); } int cigemmlt_ampere_8(Context *context, int m, int n, int k, const int8_t *A, const int8_t *B, void *C, float *row_scale, int lda, int ldb, int ldc) - { return igemmlt_ampere_8_rowscale((cublasLtHandle_t) context->m_handle, m, n, k, A, B, C, row_scale, lda, ldb, ldc); } + { return igemmlt_ampere_8((cublasLtHandle_t) context->m_handle, m, n, k, A, B, C, row_scale, lda, ldb, ldc); } #define MAKE_FUNC_CTRANSFORM(fbits, fsrc, ftrgt, ftranspose, dtype, src, target, transpose, bits) \ void ctransform_##fbits##_##fsrc##_to_##ftrgt##_##ftranspose(Context *context, dtype *A, dtype *out, int dim1, int dim2) \ |