summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorjustheuristic <justheuristic@gmail.com>2022-09-17 23:06:57 +0300
committerjustheuristic <justheuristic@gmail.com>2022-09-17 23:06:57 +0300
commit469d5a631d77d135f055d3aa012ac852a0ef0856 (patch)
treed1610ac802645b49335f74ac57fa2a2324c6abc0
parentcc4858c2fd48ef17a888b9d45bb35bb00e373eb8 (diff)
test_bf16
-rw-r--r--tests/test_autograd.py2
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(