1818# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
1919# THE SOFTWARE.
2020
21- from functools import wraps
2221from math import pi as math_pi
23- from warnings import warn
2422from triton .language import core , math , semantic , standard
2523from triton ._C .libtriton import ir
2624from triton .runtime .jit import jit
2725from triton .backends .ascend .utils import is_compile_on_910_95 , triton_enable_libdevice_simt
2826
27+ from .utils import _deprecated
28+
2929
3030def _is_libdevice_simt_enabled (_semantic ) -> bool :
3131 return triton_enable_libdevice_simt (_semantic .builder .options .arch )
@@ -35,23 +35,6 @@ def _is_a5_target(_semantic) -> bool:
3535 return is_compile_on_910_95 (_semantic .builder .options .arch )
3636
3737
38- def _deprecated (replacement ):
39-
40- def decorator (fn ):
41- message = (f"cann.libdevice.{ fn .__name__ } is deprecated and will be removed in the next release; "
42- f"use cann.libdevice.{ replacement } instead." )
43-
44- @wraps (fn )
45- def wrapper (* args , ** kwargs ):
46- warn (message , FutureWarning , stacklevel = 2 )
47- return fn (* args , ** kwargs )
48-
49- wrapper .__doc__ = f"{ fn .__doc__ or '' } \n \n .. warning::\n { message } "
50- return wrapper
51-
52- return decorator
53-
54-
5538class _FlipStaticRange :
5639
5740 def __init__ (self , arg1 , arg2 = None , step = None ):
@@ -2913,30 +2896,35 @@ def sqrt(arg0, _semantic=None):
29132896
29142897
29152898@core .extern
2916- @_deprecated ("hadd" )
2899+ @_deprecated (fn_name = "triton.language.extra.cann.libdevice.uhadd" ,
2900+ replacement = "triton.language.extra.cann.libdevice.hadd" )
29172901def uhadd (arg0 , arg1 , _semantic = None ):
29182902 return hadd (arg0 , arg1 , _semantic = _semantic )
29192903
29202904
29212905@core .extern
2922- @_deprecated ("mul24" )
2906+ @_deprecated (fn_name = "triton.language.extra.cann.libdevice.umul24" ,
2907+ replacement = "triton.language.extra.cann.libdevice.mul24" )
29232908def umul24 (arg0 , arg1 , _semantic = None ):
29242909 return mul24 (arg0 , arg1 , _semantic = _semantic )
29252910
29262911
29272912@core .extern
2928- @_deprecated ("mulhi" )
2913+ @_deprecated (fn_name = "triton.language.extra.cann.libdevice.umulhi" ,
2914+ replacement = "triton.language.extra.cann.libdevice.mulhi" )
29292915def umulhi (arg0 , arg1 , _semantic = None ):
29302916 return mulhi (arg0 , arg1 , _semantic = _semantic )
29312917
29322918
29332919@core .extern
2934- @_deprecated ("rhadd" )
2920+ @_deprecated (fn_name = "triton.language.extra.cann.libdevice.urhadd" ,
2921+ replacement = "triton.language.extra.cann.libdevice.rhadd" )
29352922def urhadd (arg0 , arg1 , _semantic = None ):
29362923 return rhadd (arg0 , arg1 , _semantic = _semantic )
29372924
29382925
29392926@core .extern
2940- @_deprecated ("sad" )
2927+ @_deprecated (fn_name = "triton.language.extra.cann.libdevice.usad" ,
2928+ replacement = "triton.language.extra.cann.libdevice.sad" )
29412929def usad (arg0 , arg1 , arg2 , _semantic = None ):
29422930 return sad (arg0 , arg1 , arg2 , _semantic = _semantic )
0 commit comments