Skip to content

Commit

Permalink
pas platform option and arg to profiling core tool
Browse files Browse the repository at this point in the history
Signed-off-by: cindyyuanjiang <[email protected]>
  • Loading branch information
cindyyuanjiang committed Dec 7, 2023
1 parent 6e8f1f5 commit b150e5f
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion user_tools/src/spark_rapids_pytools/rapids/profiling.py
Original file line number Diff line number Diff line change
Expand Up @@ -247,7 +247,8 @@ def _process_output(self):
self.__generate_report_with_recommendations()

def _init_rapids_arg_list(self) -> List[str]:
return self._create_autotuner_rapids_args()
platform_arg = ['--platform', self.ctxt.platform.get_platform_name().replace('_', '-')]
return [platform_arg] + self._create_autotuner_rapids_args()


@dataclass
Expand Down

0 comments on commit b150e5f

Please sign in to comment.