Skip to content

ROCm port#11

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

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

Conversation

@jeffdaily
Copy link

Tests verified as passing on MI200. Benchmarking TBD.

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

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
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

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

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