|
20 | 20 | Conv2D2BNInfoCollectionTest, Conv2DBNChainInfoCollectionTest, BNChainInfoCollectionTest, \ |
21 | 21 | BNLayerInfoCollectionTest, INP2BNInfoCollectionTest |
22 | 22 | from tests.pytorch_tests.function_tests.get_gptq_config_test import TestGetGPTQConfig |
23 | | -from tests.pytorch_tests.function_tests.resource_utilization_data_test import TestResourceUtilizationDataBasicAllBitwidth, \ |
24 | | - TestResourceUtilizationDataBasicPartialBitwidth, TestResourceUtilizationDataComplexPartialBitwidth, TestResourceUtilizationDataComplexAllBitwidth |
25 | 23 | from tests.pytorch_tests.function_tests.layer_fusing_test import LayerFusingTest1, LayerFusingTest2, LayerFusingTest3, \ |
26 | 24 | LayerFusingTest4 |
27 | 25 | from tests.pytorch_tests.function_tests.set_device_test import SetDeviceTest |
@@ -84,30 +82,6 @@ def test_inp2bn_bn_info_collection(self): |
84 | 82 | """ |
85 | 83 | INP2BNInfoCollectionTest(self).run_test() |
86 | 84 |
|
87 | | - def test_ru_data_basic_all(self): |
88 | | - """ |
89 | | - This test checks the resource utilization data Pytorch API. |
90 | | - """ |
91 | | - TestResourceUtilizationDataBasicAllBitwidth(self).run_test() |
92 | | - |
93 | | - def test_ru_data_basic_partial(self): |
94 | | - """ |
95 | | - This test checks the resource utilization data Pytorch API. |
96 | | - """ |
97 | | - TestResourceUtilizationDataBasicPartialBitwidth(self).run_test() |
98 | | - |
99 | | - def test_ru_data_complex_all(self): |
100 | | - """ |
101 | | - This test checks the resource utilization data Pytorch API. |
102 | | - """ |
103 | | - # TODO maxcut: test fails to fund lowest cut (3*224*250 + 3). also need to fix the "max_tensor" of the test Model. |
104 | | - TestResourceUtilizationDataComplexAllBitwidth(self).run_test() |
105 | | - |
106 | | - def test_ru_data_complex_partial(self): |
107 | | - """ |
108 | | - This test checks the resource utilization data Pytorch API. |
109 | | - """ |
110 | | - TestResourceUtilizationDataComplexPartialBitwidth(self).run_test() |
111 | 85 |
|
112 | 86 | def test_activation_hessian_trace(self): |
113 | 87 | """ |
|
0 commit comments