File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -467,7 +467,7 @@ def test_linear_fuse_relu_forward(self):
467
467
for bias in [True , False ]:
468
468
linear = torch .nn .Linear (in_features , out_features , bias = bias ).to ('dpcpp' ).to (dtype )
469
469
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 )
471
471
linear_fuse_relu .weight .data = linear .weight .clone ()
472
472
if bias :
473
473
linear_fuse_relu .bias .data = linear .bias .clone ()
@@ -486,7 +486,7 @@ def test_linear_fuse_relu_backward(self):
486
486
for bias in [True , False ]:
487
487
linear = torch .nn .Linear (in_features , out_features , bias = bias ).to ('dpcpp' ).to (dtype )
488
488
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 )
490
490
linear_fuse_relu .weight .data = linear .weight .clone ()
491
491
if bias :
492
492
linear_fuse_relu .bias .data = linear .bias .clone ()
You can’t perform that action at this time.
0 commit comments