Skip to content

Commit b5bd684

Browse files
committed
checker workaround
1 parent 98afaa0 commit b5bd684

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

onnxscript/rewriter/matmul_add_to_gemm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77
- Add(MatMul(Transpose(X), Transpose(W)), B) -> Gemm
88
"""
99

10-
from abc import ABC
10+
import abc
1111
from typing import ClassVar
1212

1313
from onnxscript.rewriter import pattern as orp
1414

1515

16-
class _MatMulAddToGemmBase(orp.RewriteRuleClassBase, ABC):
16+
class _MatMulAddToGemmBase(orp.RewriteRuleClassBase, abc.ABC):
1717
trans_a: ClassVar = False
1818
trans_b: ClassVar = False
1919

onnxscript/rewriter/matmul_add_to_gemm_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def get_test_model(
6767
b = ir.tensor(self.rng.uniform(-0.5, 0.5, bias_shape).astype("float32"), name="B")
6868
b = tape.initializer(b)
6969

70+
x_t, w_t = None, None
7071
if transA:
7172
x_t = tape.op("Transpose", inputs=[x], attributes={"perm": permA})
7273

0 commit comments

Comments
 (0)