diff options
author | justheuristic <justheuristic@gmail.com> | 2022-09-20 06:51:25 +0300 |
---|---|---|
committer | justheuristic <justheuristic@gmail.com> | 2022-09-20 06:51:25 +0300 |
commit | 76ce9aa6da7d68d2463f0f3e99532ab5b6db58a8 (patch) | |
tree | 75454081c86ba1c96c07e83defc9fc5f4de840cf | |
parent | 292a47871603cc1ebe620221358d571a8f5c6d8f (diff) |
try fp32
-rw-r--r-- | tests/test_autograd.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_autograd.py b/tests/test_autograd.py index cb82898..40bb441 100644 --- 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( |