From 0d35990eaafaacaff39da0308da62dc03dbe1911 Mon Sep 17 00:00:00 2001 From: wangyibo <1605891897@qq.com> Date: Tue, 3 Jun 2025 17:09:48 +0800 Subject: [PATCH] add ut test --- .jenkins/test/config/dependent_packages.yaml | 2 +- tools/rules/line_rules.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.jenkins/test/config/dependent_packages.yaml b/.jenkins/test/config/dependent_packages.yaml index 72069bf7..39ca2851 100644 --- a/.jenkins/test/config/dependent_packages.yaml +++ b/.jenkins/test/config/dependent_packages.yaml @@ -1,2 +1,2 @@ mindspore: - 'https://repo.mindspore.cn/mindspore/mindspore/version/202505/20250526/master_20250526010017_b6f48709740e52a1daf206553f742609a0998bae_newest/' \ No newline at end of file + 'https://repo.mindspore.cn/mindspore/mindspore/version/202505/20250529/master_20250529010018_c66603449f1452339350f621ebf006fcedae187b_newest/' \ No newline at end of file diff --git a/tools/rules/line_rules.py b/tools/rules/line_rules.py index fe479623..b21ca2d7 100644 --- a/tools/rules/line_rules.py +++ b/tools/rules/line_rules.py @@ -136,7 +136,7 @@ LINE_RULES = { + # run backward + grad_ = C.GradOperation(True, True, True) + weights = model.trainable_params() -+ _pynative_executor.check_run(grad_, config.forward_step_func, weights, None, input_tensor[0]) ++ _pynative_executor.check_run(grad_, config.forward_step_func, weights, None, input_tensor[0], create_graph=False) + _pynative_executor.grad(config.forward_step_func, grad_, weights, None, input_tensor[0], output_tensor_grad[0]) # Collect the grad of the input_tensor. -- Gitee