Skip to content

Commit ff837ca

Browse files
authored
Update test_rnn_relu.py
1 parent 0256ff7 commit ff837ca

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

benchmark/test_rnn_relu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def test_rnn_relu():
6060
# software conversion for every load/store, which makes the RNN
6161
# hidden-state recurrence ~20x slower than PyTorch's cuDNN path.
6262
# Replace bf16 with fp16 on these devices so the benchmark still
63-
# exercises 3 dtype entries with meaningful numbers.
63+
# exercises 3 dtype entries with meaningful number.
6464
major, _ = torch.cuda.get_device_capability()
6565
if major < 8 and torch.bfloat16 in dtypes:
6666
dtypes[dtypes.index(torch.bfloat16)] = torch.float16

0 commit comments

Comments
 (0)