Skip to content

Commit f95664c

Browse files
committed
make tensor core path available for cc 7.5 and above
1 parent 417cfc3 commit f95664c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cuda/conv2d-implicit.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -927,7 +927,7 @@ static void conv2d_implicit_cuda(const float * X_D, const T * K_D, float * Y_D,
927927

928928
static void conv2d_implicit_cuda_f16(ggml_backend_cuda_context & ctx, const float * X_D, const half * K_D, float * Y_D, int cc, const param_t P, cudaStream_t st) {
929929

930-
if (GGML_CUDA_CC_IS_NVIDIA(cc) && ampere_mma_available(cc) && P.c % 8 == 0 && (P.r > 1 || P.s > 1)) {
930+
if (GGML_CUDA_CC_IS_NVIDIA(cc) && turing_mma_available(cc) && P.c % 8 == 0 && (P.r > 1 || P.s > 1)) {
931931

932932
int id = ggml_cuda_get_device();
933933

0 commit comments

Comments
 (0)