Move optical flow tests from specific TU test job to Ampere tests

Signed-off-by: Joaquin Anton Guirao <janton@nvidia.com>
NVIDIA · Jan 9, 2025 · c32b46c · c32b46c
1 parent 3f5646a
commit c32b46c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 18 deletions.
diff --git a/qa/TL0_python_self_test_TU/test.sh b/qa/TL0_python_self_test_TU/test.sh
diff --git a/qa/TL0_self_test_Ampere/test.sh b/qa/TL0_self_test_Ampere/test.sh
@@ -31,6 +31,10 @@ test_body() {
 
   # test decoders on A100 as well
   ${python_new_invoke_test} -s decoder test_image
+
+  # test Optical Flow
+  ${python_invoke_test} test_optical_flow.py
+  ${python_invoke_test} test_dali_variable_batch_size.py:test_optical_flow
 }
 
 pushd ../..