Reduce the workspace for TRT index_select UT (#56069)

597f6d43 · Leo Chen · GitHub · 242d26d4 · 597f6d43
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

test/ir/inference/test_trt_convert_index_select.py test/ir/inference/test_trt_convert_index_select.py +1 -1

未找到文件。
--- a/test/ir/inference/test_trt_convert_index_select.py
+++ b/test/ir/inference/test_trt_convert_index_select.py
@@ -191,7 +191,7 @@ class TrtConvertIndexSelectTest(TrtLayerAutoScanTest):
        yield self.create_inference_config(), generate_trt_nodes_num(True), 1e-3

    def test(self):
-        self.trt_param.workspace_size = 1 << 60
+        self.trt_param.workspace_size = 1 << 10
        self.run_test()