Skip to content

Conversation

@t-vi
Copy link
Contributor

@t-vi t-vi commented Sep 10, 2020

This fixes the ROCm backend after #6347 to also allow mcpu and mtriple attributes for ROCm target objects.

With this change, we can run the gpu_imagenet_bench.py again.

@junrushao1994 @masahi

@masahi
Copy link
Member

masahi commented Sep 10, 2020

Nice, just today I did a clean install of rocm 3.7, and wondered why the benchmark is broken. So timely for me :)

@tqchen
Copy link
Member

tqchen commented Sep 10, 2020

@t-vi please rebase against the master to resolve conflicts

@junrushao
Copy link
Member

Hmmm I think the issue has been fixed by #6369

@t-vi
Copy link
Contributor Author

t-vi commented Sep 10, 2020

@junrushao1994 Right, thank you! @masahi @tqchen sorry for the noise. I had searched for rocm/mcpu patches but missed the more general one.

@t-vi t-vi closed this Sep 10, 2020
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