add test_tensor_wrapper into CI steps

FlagOpen · Sep 11, 2024 · d0c0c52 · d0c0c52
1 parent f6c8052
commit d0c0c52
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/op-unit-test.sh b/tools/op-unit-test.sh
@@ -29,6 +29,7 @@ cmds=(
    "CUDA_VISIBLE_DEVICES=6 coverage run ${COVERAGE_ARGS} -m pytest -s tests/test_libentry.py &"
    "CUDA_VISIBLE_DEVICES=6 coverage run ${COVERAGE_ARGS} -m pytest -s tests/test_pointwise_dynamic.py &"
    "CUDA_VISIBLE_DEVICES=6 coverage run ${COVERAGE_ARGS} -m pytest -s tests/test_shape_utils.py &"
+   "CUDA_VISIBLE_DEVICES=6 coverage run ${COVERAGE_ARGS} -m pytest -s tests/test_tensor_wrapper.py &"
 )
 
 declare -a exit_statuses