Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ROCm] Optimize ROCm CI pipeline 2 #16691

Merged
merged 6 commits into from
Jul 24, 2023
Merged

Conversation

PeixuanZuo
Copy link
Contributor

@PeixuanZuo PeixuanZuo commented Jul 13, 2023

  • Set KERNEL_EXPLORER_TEST_USE_CUPY=1 to replace numpy with cupy on kernel explorer test.

KERNEL_EXPLORER_TEST_USE_CUPY=0 The CPU utilization is shown as below:
image

KERNEL_EXPLORER_TEST_USE_CUPY=1 The CPU utilization is shown as below:
image

  • Use Bash@3.
  • Update shell script.

@PeixuanZuo PeixuanZuo requested a review from a team as a code owner July 13, 2023 10:16
@PeixuanZuo PeixuanZuo requested review from cloudhan and ytaous July 13, 2023 10:16
@snnn
Copy link
Member

snnn commented Jul 20, 2023

@PeixuanZuo, I'm OOF now. You may ask Yi to help you review pipeline changes like this whether I'm at work or not. He is in the same time zone as you.

@PeixuanZuo PeixuanZuo requested a review from mszhanyi July 21, 2023 07:35
@mszhanyi
Copy link
Contributor

It's better to list the gains after the optimization.

@PeixuanZuo PeixuanZuo merged commit 8ede2f1 into main Jul 24, 2023
@PeixuanZuo PeixuanZuo deleted the peixuanzuo/use_cupy_on_pipeline branch July 24, 2023 05:57
jchen351 pushed a commit that referenced this pull request Aug 12, 2023
- Set `KERNEL_EXPLORER_TEST_USE_CUPY=1` to replace numpy with cupy on
kernel explorer test.

KERNEL_EXPLORER_TEST_USE_CUPY=0 The CPU utilization is shown as below:

![image](https://github.com/microsoft/onnxruntime/assets/94887879/91724b78-0b4e-4cbd-ad88-83cad9976472)

KERNEL_EXPLORER_TEST_USE_CUPY=1 The CPU utilization is shown as below:

![image](https://github.com/microsoft/onnxruntime/assets/94887879/58239911-667c-4d5f-bb78-deca60d0266f)


- Use `Bash@3`.
- Update shell script.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants