Skip to content

Commit ff10ff4

Browse files
committed
Merge branch 'anuragd/enable_tests' into 'release/1.0'
(//tests): Enabled the test cases removed from DLFW See merge request adlsa/TRTorch!12
2 parents db49786 + 3ccca83 commit ff10ff4

File tree

4 files changed

+13
-14
lines changed

4 files changed

+13
-14
lines changed

tests/core/conversion/converters/BUILD

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,14 +145,15 @@ test_suite(
145145
":test_interpolate",
146146
":test_layer_norm",
147147
":test_linear",
148-
#":test_lstm_cell",
148+
":test_lstm_cell",
149149
":test_matrix_multiply",
150150
":test_normalize",
151151
":test_pooling",
152152
":test_reduce",
153153
":test_replication_pad",
154154
":test_select",
155155
":test_shuffle",
156+
":test_softmax",
156157
":test_squeeze",
157158
":test_stack",
158159
":test_topk",

tests/core/conversion/converters/test_activation.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,6 @@ TEST(Converters, ATenEluConvertsCorrectly) {
200200
ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 2e-6));
201201
}
202202

203-
#if 0
204203
TEST(Converters, ATenGELUConvertsCorrectly) {
205204
const auto graph = R"IR(
206205
graph(%0 : Tensor):
@@ -227,4 +226,3 @@ TEST(Converters, ATenGELUConvertsCorrectly) {
227226

228227
ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0], 5e-3));
229228
}
230-
#endif

tests/core/partitioning/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ test_suite(
8484
":test_shape_analysis",
8585
":test_tensorrt_conversion",
8686
":test_stitched_graph",
87-
#":test_fallback_graph_output",
87+
":test_fallback_graph_output",
8888
":test_loop_fallback",
8989
":test_conditionals"
9090
]

tests/cpp/BUILD

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,12 @@ test_suite(
1212
name = "api_tests",
1313
tests = [
1414
":test_default_input_types",
15-
#":test_compiled_modules",
16-
#":test_modules_as_engines",
17-
#":test_runtime_thread_safety",
18-
#":test_multiple_registered_engines",
15+
":test_compiled_modules",
16+
":test_modules_as_engines",
17+
":test_runtime_thread_safety",
18+
":test_multiple_registered_engines",
1919
":test_serialization",
20-
#":test_module_fallback",
20+
":test_module_fallback",
2121
":test_example_tensors"
2222
],
2323
)
@@ -26,12 +26,12 @@ test_suite(
2626
name = "aarch64_api_tests",
2727
tests = [
2828
":test_default_input_types",
29-
#":test_compiled_modules",
30-
#":test_modules_as_engines",
31-
#":test_runtime_thread_safety",
32-
#":test_multiple_registered_engines",
29+
":test_compiled_modules",
30+
":test_modules_as_engines",
31+
":test_runtime_thread_safety",
32+
":test_multiple_registered_engines",
3333
":test_serialization",
34-
#":test_module_fallback",
34+
":test_module_fallback",
3535
":test_example_tensors"
3636
],
3737
)

0 commit comments

Comments
 (0)