diff --git a/torchbenchmark/util/triton_op.py b/torchbenchmark/util/triton_op.py index 19aaf54c5e..2e3c5e3505 100644 --- a/torchbenchmark/util/triton_op.py +++ b/torchbenchmark/util/triton_op.py @@ -878,7 +878,8 @@ def ncu_trace(self, input_id: int, fn_name: str, replay: bool=False, profile_ir= import sys import subprocess - op_task_args = copy.deepcopy(sys.argv) + op_task_args = [] if IS_FBCODE else [sys.executable] + op_task_args.extend(copy.deepcopy(sys.argv)) for override_option in ["--only", "--input-id", "--num-inputs", "--metrics"]: op_task_args = _remove_params( op_task_args, _find_param_loc(op_task_args, override_option)