Lazy scipy import (#2250)

This commit is contained in:
Michael Lazos
2025-04-30 16:10:00 -04:00
committed by GitHub
parent b3ce7e12b7
commit c4bdfe821c
+1 -1
View File
@@ -34,7 +34,6 @@ import ctypes
from cutlass_library import SubstituteTemplate
import numpy as np
from scipy.special import erf
from cutlass_library import DataType, DataTypeTag
from cutlass.backend.c_types import MatrixCoord_, tuple_factory
@@ -530,6 +529,7 @@ class hardswish(ActivationFunctor, metaclass=hardswishMeta):
class geluMeta(ActivationMeta):
@classmethod
def numpy(cls, x):
from scipy.special import erf
return 0.5 * x * (1 + erf(x / np.sqrt(2.0)))
@classmethod