Skip to content

Commit 54e67af

Browse files
committed
Modify the variable replace in the for loop to inplace in tests/test_relu.py
1 parent 1015917 commit 54e67af

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/test_relu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,9 @@ def test_cuda(shape, dtype, atol, rtol):
1414

1515
input = torch.randn(shape, dtype=dtype, device=device)
1616

17-
for replace in (False, True):
18-
ninetoothed_output = ntops.torch.relu(input, replace)
19-
reference_output = F.relu(input, replace)
17+
for inplace in (False, True):
18+
ninetoothed_output = ntops.torch.relu(input, inplace)
19+
reference_output = F.relu(input, inplace)
2020

2121
assert torch.allclose(
2222
ninetoothed_output, reference_output, atol=atol, rtol=rtol

0 commit comments

Comments
 (0)