add

FlagOpen · Aug 22, 2024 · b5c3231 · b5c3231
1 parent c576ecb
commit b5c3231
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tools/code_coverage/coverage.sh b/tools/code_coverage/coverage.sh
@@ -13,7 +13,7 @@ FlagGemsROOT="$( cd "$( dirname "${BASH_SOURCE[0]}")/../../" && pwd )"
 echo ${FlagGemsROOT}
 
 cmds=(
-   "CUDA_VISIBLE_DEVICES=0 coverage run --parallel-mode --omit "*/.flaggems/*","*/usr/lib/*" --source=./src,./tests -m pytest -s tests/test_unary_pointwise_ops.py &"
+   "CUDA_VISIBLE_DEVICES=0 coverage run --parallel-mode --omit "*/.flaggems/*","*/usr/lib/*" --source=./src,./tests,./tools/code_coverage -m pytest -s tests/test_unary_pointwise_ops.py::test_accuracy_abs &"
 #    "CUDA_VISIBLE_DEVICES=0 coverage run --parallel-mode --omit "*/.flaggems/*","*/usr/lib/*" --source=./src,./tests -m pytest -s tests/test_pointwise_type_promotion.py &"
 #    "CUDA_VISIBLE_DEVICES=1 coverage run --parallel-mode --omit "*/.flaggems/*","*/usr/lib/*" --source=./src,./tests -m pytest -s tests/test_binary_pointwise_ops.py &"
 #    "CUDA_VISIBLE_DEVICES=1 coverage run --parallel-mode --omit "*/.flaggems/*","*/usr/lib/*" --source=./src,./tests -m pytest -s tests/test_tensor_constructor_ops.py &"

diff --git a/tools/code_coverage/jit_func_position.py b/tools/code_coverage/jit_func_position.py
@@ -69,6 +69,8 @@ def get_end_line(self, node):
     args = parser.parse_args()
 
     for filename in args.files:
+        if not filename.endswith(".py"):
+            continue
         logging.debug(filename)
         with open(filename, "r") as file:
             source_code = file.read()