From c4bdfe821c6d017122e670cbbbab36dcd32691ea Mon Sep 17 00:00:00 2001 From: Michael Lazos Date: Wed, 30 Apr 2025 13:10:00 -0700 Subject: [PATCH] Lazy scipy import (#2250) --- python/cutlass/backend/epilogue.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cutlass/backend/epilogue.py b/python/cutlass/backend/epilogue.py index a9cf6be7..d7f2bd62 100644 --- a/python/cutlass/backend/epilogue.py +++ b/python/cutlass/backend/epilogue.py @@ -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