Skip to content

Commit 76ce9aa

Browse files
committed
try fp32
1 parent 292a478 commit 76ce9aa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_autograd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def test_matmul(dim1, dim2, dim3, dim4, funcs, dtype, req_grad, transpose):
253253

254254
transpose = [(False, True), (False, False)]
255255
str_transpose = ["NT", "NN"]
256-
dtype = [torch.float16, torch.bfloat16]
256+
dtype = [torch.float16, torch.bfloat16, torch.float32]
257257
has_fp16_weights = [True, False]
258258
has_bias = [True, False]
259259
values = list(

0 commit comments

Comments
 (0)