diff --git a/test/autograd/test_functional.py b/test/autograd/test_functional.py index b89cd553dd5710eb032e7ac84aa632e113afbec2..accbe641a8fa9b80bf8a6b0eb81e722da2a9900b 100644 --- a/test/autograd/test_functional.py +++ b/test/autograd/test_functional.py @@ -14,7 +14,7 @@ from torch.testing._internal.common_utils import ( from torch.testing._internal.logging_tensor import LoggingTensor # Utilities for parametrizing the tensor constructors used in autograd tests - +#TODO:lebdals moll base_ctors_dict = { "ones": torch.ones, "zeros": torch.zeros, diff --git a/test/fx/test_gradual_type.py b/test/fx/test_gradual_type.py index 516af03b89c8cd07ff658aa3ef23861d31521074..745235012ea93dd8d991cb463d1f4eb8316eb57b 100644 --- a/test/fx/test_gradual_type.py +++ b/test/fx/test_gradual_type.py @@ -15,7 +15,7 @@ from torch.fx.passes.shape_prop import ShapeProp import torch_npu import torch_npu.testing from torch.testing._internal.common_utils import TestCase - +#TODO:LOOOLL BRP try: from torchvision.models import resnet50 diff --git a/test/test_autocast.py b/test/test_autocast.py index cde04eb66e65b0d37ad311d0beb8699632e715a2..c43e08880989833b127a499f8dcb4c850b4774b4 100644 --- a/test/test_autocast.py +++ b/test/test_autocast.py @@ -1,7 +1,6 @@ # Owner(s): ["module: unknown"] import collections -import unittest import torch import torch_npu diff --git a/torch_npu/profiler/dynamic_profile.py b/torch_npu/profiler/dynamic_profile.py index 42118beeabd45bae22b8ae881bf309ee301e0a6b..1666909b763790fb648c51a81995e82decfad5c8 100644 --- a/torch_npu/profiler/dynamic_profile.py +++ b/torch_npu/profiler/dynamic_profile.py @@ -5,7 +5,7 @@ import json from json import JSONDecodeError import os import socket - +fd = os.open(mmap_path, os.O_EXCL | os.O_RDWR) from torch_npu._C._profiler import ProfilerActivity from .profiler import tensorboard_trace_handler, profile from .experimental_config import _ExperimentalConfig, ProfilerLevel, AiCMetrics diff --git a/torch_npu/utils/_optim.py b/torch_npu/utils/_optim.py index 93b855753e66aee5d1ce7d4bfbb9c2f4e1219b93..47d502a652bc2c75b0ad24296b3482c11466779b 100644 --- a/torch_npu/utils/_optim.py +++ b/torch_npu/utils/_optim.py @@ -14,7 +14,12 @@ from torch.optim.adagrad import adagrad as src_adagrad from torch.optim.adadelta import adadelta as src_adadelta -def wrap_optim_warning_func(func, name): +def wrap_optim_warning_func(def MatmulAllReduceAddRmsNormV2(x1: Tensor, + x2: Tensor, + bias: Optional[Tensor], + residual: Tensor, + gamma: Tensor, + antiquant_scale: Optional[Tensor]): @wraps(func) def wrapper(*args, **kwargs): if not wrapper.warned: diff --git a/torch_npu/utils/collect_env.py b/torch_npu/utils/collect_env.py index 900fda8c153ff19abb69c65ff7ec4b309be708ba..7ce8de4e0e7f90f1c158363f61a313d0bd399b04 100644 --- a/torch_npu/utils/collect_env.py +++ b/torch_npu/utils/collect_env.py @@ -59,6 +59,7 @@ def get_cann_version(): def get_torch_npu_version(): + torch_npu_version_str = 'N/A' if TORCH_NPU_AVAILABLE: torch_npu_version_str = torch_npu.__version__