diff --git a/torch_npu/utils/_dynamo.py b/torch_npu/utils/_dynamo.py index 5915b8ed9c96a8b2d46a1cc94a31de505dcc24a4..c40a300e2eb67ce8ce411a912539edb2f7782128 100644 --- a/torch_npu/utils/_dynamo.py +++ b/torch_npu/utils/_dynamo.py @@ -76,6 +76,7 @@ def UserDefinedClassVariable__new__(cls, value, **kwargs): return cls.__new__raw(cls) + def SkipFunctionVariable__new__(cls, value, reason=None, **kwargs): if value in [ torch.npu.stream,