There was an error while loading. Please reload this page.
1 parent 0256ff7 commit ff837caCopy full SHA for ff837ca
1 file changed
benchmark/test_rnn_relu.py
@@ -60,7 +60,7 @@ def test_rnn_relu():
60
# software conversion for every load/store, which makes the RNN
61
# hidden-state recurrence ~20x slower than PyTorch's cuDNN path.
62
# Replace bf16 with fp16 on these devices so the benchmark still
63
- # exercises 3 dtype entries with meaningful numbers.
+ # exercises 3 dtype entries with meaningful number.
64
major, _ = torch.cuda.get_device_capability()
65
if major < 8 and torch.bfloat16 in dtypes:
66
dtypes[dtypes.index(torch.bfloat16)] = torch.float16
0 commit comments