leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 28 Packages Projects Releases Wiki Activity

Lazy scipy import (#2250)

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