Fix name 'np' is not defined error

Signed-off-by: Molly Sophia <[email protected]>
inisis · Nov 23, 2024 · a46514d · a46514d
1 parent 31ce219
commit a46514d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnxslim/core/pattern/fusion/gemm.py b/onnxslim/core/pattern/fusion/gemm.py
@@ -1,3 +1,5 @@
+import numpy as np
+
 import onnxslim.third_party.onnx_graphsurgeon as gs
 from onnxslim.core.optimization.dead_node_elimination import get_constant_variable
 from onnxslim.core.pattern import Pattern, PatternMatcher, get_node_users