Drop nvtx test (#19154)

tests/run_standalone_tests.sh

@@ -31,7 +31,7 @@ test_path=$1
 printf "source path: $test_path\n"
 
 # collect all tests with parametrization based filtering with PL_RUN_STANDALONE_TESTS
-standalone_tests=$(python -m pytest $test_path -q --collect-only --pythonwarnings ignore)
+standalone_tests=$(python3 -m pytest $test_path -q --collect-only --pythonwarnings ignore)
 printf "Collected tests: \n $standalone_tests"
 # match only lines with tests
 parametrizations=$(grep -oP '\S+::test_\S+' <<< "$standalone_tests")
@@ -69,7 +69,7 @@ for i in "${!parametrizations_arr[@]}"; do
   # execute the test in the background
   # redirect to a log file that buffers test output. since the tests will run in the background, we cannot let them
   # output to std{out,err} because the outputs would be garbled together
-  python ${defaults} "$parametrization" &>> standalone_test_output.txt &
+  python3 ${defaults} "$parametrization" &>> standalone_test_output.txt &
   # save the PID in an array
   pids[${i}]=$!
   # add row to the final report

tests/tests_pytorch/profilers/test_profiler.py

@@ -471,24 +471,6 @@ def look_for_trace(trace_dir):
     assert look_for_trace(tmpdir / "lightning_logs" / "version_0")
 
 
-# Todo: this test has not been running as all our CI GPU runners have higher capacity
-# @RunIf(min_cuda_gpus=1, standalone=True)
-# @pytest.mark.skipif(torch.cuda.get_device_capability()[0] >= 8)
-# def test_pytorch_profiler_nested_emit_nvtx():
-#     """This test check emit_nvtx is correctly supported."""
-#     profiler = PyTorchProfiler(use_cuda=True, emit_nvtx=True)
-#     model = BoringModel()
-#     trainer = Trainer(
-#         fast_dev_run=True,
-#         profiler=profiler,
-#         accelerator="gpu",
-#         devices=1,
-#         enable_progress_bar=False,
-#         enable_model_summary=False,
-#     )
-#     trainer.fit(model)
-
-
 def test_register_record_function(tmpdir):
     use_cuda = torch.cuda.is_available()
     pytorch_profiler = PyTorchProfiler(

tests/tests_pytorch/run_standalone_tasks.sh

@@ -18,17 +18,6 @@ set -e
 # this environment variable allows special tests to run
 export PL_RUN_STANDALONE_TESTS=1
 
-#can_run_nvprof=$(python -c "import torch; print(torch.cuda.is_available() and torch.cuda.get_device_capability()[0] < 8)")
-#if [[ $can_run_nvprof == "True" ]]; then
-#    echo "Running profilers/test_profiler.py::test_pytorch_profiler_nested_emit_nvtx"
-#    nvprof --profile-from-start off \
-#      -o trace_name.prof \
-#        -- python -m coverage run \
-#          --source lightning.pytorch \
-#          --append -m pytest \
-#            --no-header profilers/test_profiler.py::test_pytorch_profiler_nested_emit_nvtx
-#fi
-
 # test that a user can manually launch individual processes
 echo "Running manual ddp launch test"
 export PYTHONPATH="${PYTHONPATH}:$(pwd)"