Skip to content

Commit

Permalink
fix the comment of test
Browse files Browse the repository at this point in the history
  • Loading branch information
Wanglongzhi2001 committed Nov 26, 2023
1 parent e7c6751 commit 267f5f1
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def get_cuda_version():

@unittest.skipIf(
core.is_compiled_with_cuda() and get_cuda_version() < 11020,
"weight_only_linear needs CUDA version greater than 11.2",
"weight_only_linear needs CUDA version greater or euqal than 11.2",
)
class TestMatmulToWeightOnly(unittest.TestCase):
def test_matmul_to_weight_only(self):
Expand Down

0 comments on commit 267f5f1

Please sign in to comment.