try fp32

76ce9aa6 · justheuristic · 292a4787 · 76ce9aa6
Commit 76ce9aa6 authored Sep 20, 2022 by justheuristic
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_autograd.py tests/test_autograd.py +1 -1

No files found.
--- a/tests/test_autograd.py
+++ b/tests/test_autograd.py
@@ -253,7 +253,7 @@ for c in req_grad:

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