Skip to content

Commit b90c743

Browse files
authored
remove Numpy-2.0 incompatible use of numpy.product
Differential Revision: D81945242 Pull Request resolved: pytorch#14081
1 parent 3db7900 commit b90c743

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/cadence/aot/tests/test_ref_implementations.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -216,12 +216,12 @@ def test_quantized_linear(
216216
expected_output: torch.Tensor,
217217
) -> None:
218218
src = (
219-
torch.arange(np.product(src_shape))
219+
torch.arange(np.prod(src_shape))
220220
.reshape(src_shape)
221221
.to(expected_output.dtype)
222222
)
223223
weight = (
224-
torch.arange(np.product(weight_shape))
224+
torch.arange(np.prod(weight_shape))
225225
.reshape(weight_shape)
226226
.to(expected_output.dtype)
227227
)

0 commit comments

Comments
 (0)