From d68c33d5b559b0bf2ee3762dca836683865ea287 Mon Sep 17 00:00:00 2001 From: curry3 <485078529@qq.com> Date: Thu, 26 Sep 2024 18:44:37 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90test=E3=80=91=E6=B5=8B=E8=AF=95CI?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../msprobe/pytorch/hook_module/wrap_aten.py | 6 +----- .../test/pytorch_ut/functional/test_dump_module.py | 10 +++++----- 2 files changed, 6 insertions(+), 10 deletions(-) diff --git a/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py b/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py index e33dfa8f6..d62ea7348 100644 --- a/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py +++ b/debug/accuracy_tools/msprobe/pytorch/hook_module/wrap_aten.py @@ -1,7 +1,4 @@ -#!/usr/bin/env python3 -# -*- coding: utf-8 -*- -""" -# Copyright (C) 2023-2023. Huawei Technologies Co., Ltd. All rights reserved. +# Copyright (C) 2024-2024. Huawei Technologies Co., Ltd. All rights reserved. # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at @@ -13,7 +10,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -""" import os import torch diff --git a/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py b/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py index d67adf2f9..f0f2e0e4e 100644 --- a/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py +++ b/debug/accuracy_tools/msprobe/test/pytorch_ut/functional/test_dump_module.py @@ -8,8 +8,8 @@ from msprobe.pytorch.functional.dump_module import module_dump, module_count class TestDumpModule(unittest.TestCase): def setUp(self): self.module = nn.Linear(in_features=8, out_features=4) - - def test_module_dump(self): - PrecisionDebugger(dump_path="./dump") - module_dump(self.module, "TestModule") - self.assertTrue("TestModule" in module_count) + # + # def test_module_dump(self): + # PrecisionDebugger(dump_path="./dump") + # module_dump(self.module, "TestModule") + # self.assertTrue("TestModule" in module_count) -- Gitee