未验证 提交 fba54dde 编写于 作者: C carryyu 提交者: GitHub

Fix unit test in A10 GPU (#46450)

*  Disable TF32 to solve accuracy for test_trt_conv_pass and test_trt_deformable_conv in A10 GPU.
上级 e8b9ae20
...@@ -22,6 +22,8 @@ import paddle.fluid.core as core ...@@ -22,6 +22,8 @@ import paddle.fluid.core as core
from paddle.fluid.core import PassVersionChecker from paddle.fluid.core import PassVersionChecker
from paddle.fluid.core import AnalysisConfig from paddle.fluid.core import AnalysisConfig
os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
class TensorRTSubgraphPassConvTest(InferencePassTest): class TensorRTSubgraphPassConvTest(InferencePassTest):
......
...@@ -14,6 +14,7 @@ ...@@ -14,6 +14,7 @@
from __future__ import print_function from __future__ import print_function
import os
import unittest import unittest
import numpy as np import numpy as np
from inference_pass_test import InferencePassTest from inference_pass_test import InferencePassTest
...@@ -22,6 +23,8 @@ import paddle.fluid.core as core ...@@ -22,6 +23,8 @@ import paddle.fluid.core as core
from paddle.fluid.core import PassVersionChecker from paddle.fluid.core import PassVersionChecker
from paddle.fluid.core import AnalysisConfig from paddle.fluid.core import AnalysisConfig
os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
class TRTDeformableConvTest(InferencePassTest): class TRTDeformableConvTest(InferencePassTest):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册