未验证 提交 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
from paddle.fluid.core import PassVersionChecker
from paddle.fluid.core import AnalysisConfig
os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
class TensorRTSubgraphPassConvTest(InferencePassTest):
......
......@@ -14,6 +14,7 @@
from __future__ import print_function
import os
import unittest
import numpy as np
from inference_pass_test import InferencePassTest
......@@ -22,6 +23,8 @@ import paddle.fluid.core as core
from paddle.fluid.core import PassVersionChecker
from paddle.fluid.core import AnalysisConfig
os.environ['NVIDIA_TF32_OVERRIDE'] = '0'
class TRTDeformableConvTest(InferencePassTest):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册