Skip to content

Commit

Permalink
Update linalg.cu
Browse files Browse the repository at this point in the history
Missed one
  • Loading branch information
dentalfloss1 authored Nov 14, 2023
1 parent 66cac67 commit 4a6248c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/linalg.cu
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 +209,7 @@ BFstatus bfMatMul_aa_exec(BFlinalg handle,
//bool use_bf_cherk = use_bf_cherk_str && atoi(use_bf_cherk_str);
enum { BF_CUBLAS_CHERK_THRESHOLD = 896 };
if( //use_bf_cherk &&
(CUDART_VERSION < 8000 || n < BF_CUBLAS_CHERK_THRESHOLD) &&
n < BF_CUBLAS_CHERK_THRESHOLD &&
trans == CUBLAS_OP_N &&
n % 2 == 0 &&
a_stride % 2 == 0 && a_batchstride % 2 == 0 &&
Expand Down

0 comments on commit 4a6248c

Please sign in to comment.