diff options
| author | justheuristic <justheuristic@gmail.com> | 2022-09-17 23:06:57 +0300 |
|---|---|---|
| committer | justheuristic <justheuristic@gmail.com> | 2022-09-17 23:06:57 +0300 |
| commit | 469d5a631d77d135f055d3aa012ac852a0ef0856 (patch) | |
| tree | d1610ac802645b49335f74ac57fa2a2324c6abc0 | |
| parent | cc4858c2fd48ef17a888b9d45bb35bb00e373eb8 (diff) | |
test_bf16
| -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 bae26de..05da6ed 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] +dtype = [torch.float16, torch.bfloat16] has_fp16_weights = [True, False] has_bias = [True, False] values = list( |