Skip to content

GitLab

Commit 76ce9aa6 authored by justheuristic's avatar justheuristic
Browse files

try fp32

parent 292a4787
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_autograd.py
View file @ 76ce9aa6
...@@ -253,7 +253,7 @@ for c in req_grad: ...@@ -253,7 +253,7 @@ for c in req_grad:
transpose = [(False, True), (False, False)] transpose = [(False, True), (False, False)]
str_transpose = ["NT", "NN"] str_transpose = ["NT", "NN"]
dtype = [torch.float16, torch.bfloat16] dtype = [torch.float16, torch.bfloat16, torch.float32]
has_fp16_weights = [True, False] has_fp16_weights = [True, False]
has_bias = [True, False] has_bias = [True, False]
values = list( values = list(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment