Skip to content

Commit fb9f850

Browse files
authored
Update test_lazy_reorder.py
No intel_pytorch_extension_py at release version.
1 parent 9cd8f30 commit fb9f850

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/cpu/test_lazy_reorder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -467,7 +467,7 @@ def test_linear_fuse_relu_forward(self):
467467
for bias in [True, False]:
468468
linear = torch.nn.Linear(in_features, out_features, bias=bias).to('dpcpp').to(dtype)
469469
relu = torch.nn.ReLU()
470-
linear_fuse_relu = intel_pytorch_extension_py.LinearFuseRelu(in_features, out_features, bias=bias)
470+
linear_fuse_relu = intel_pytorch_extension.LinearFuseRelu(in_features, out_features, bias=bias)
471471
linear_fuse_relu.weight.data = linear.weight.clone()
472472
if bias:
473473
linear_fuse_relu.bias.data = linear.bias.clone()
@@ -486,7 +486,7 @@ def test_linear_fuse_relu_backward(self):
486486
for bias in [True, False]:
487487
linear = torch.nn.Linear(in_features, out_features, bias=bias).to('dpcpp').to(dtype)
488488
relu = torch.nn.ReLU()
489-
linear_fuse_relu = intel_pytorch_extension_py.LinearFuseRelu(in_features, out_features, bias=bias)
489+
linear_fuse_relu = intel_pytorch_extension.LinearFuseRelu(in_features, out_features, bias=bias)
490490
linear_fuse_relu.weight.data = linear.weight.clone()
491491
if bias:
492492
linear_fuse_relu.bias.data = linear.bias.clone()

0 commit comments

Comments
 (0)