Skip to content

ROCm port#11

Open
jeffdaily wants to merge 2 commits into
Dao-AILab:masterfrom
jeffdaily:rocm
Open

ROCm port#11
jeffdaily wants to merge 2 commits into
Dao-AILab:masterfrom
jeffdaily:rocm

Conversation

@jeffdaily

Copy link
Copy Markdown

Tests verified as passing on MI200. Benchmarking TBD.

@jeffdaily jeffdaily changed the title Rocm ROCm port Nov 19, 2024

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I had to also add:

#ifndef USE_ROCM
#include <cuda_bf16.h>
#endif

Copy link
Copy Markdown
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ah, must be dependent on rocm version. I was using 6.2 and didn't need to add that.

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ah yes, I'm using 6.1.3 currently

@fanshao123456

Copy link
Copy Markdown

How long does the compilation usually take? Is it normal for it to stay stuck like this?

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.

3 participants