| #ifdef TEST_ON_CUDA |
| #include <mma.h> |
|
|
| #include <cuda_fp16.h> |
| #include <cuda_fp8.h> |
|
|
| namespace wmma = nvcuda::wmma; |
|
|
| #define LIB_CALL(call) \ |
| do { \ |
| cudaError_t err = call; \ |
| if (err != cudaSuccess) { \ |
| abort(); \ |
| } \ |
| } while (0) |
|
|
| #define HOST_TYPE(x) cuda##x |
|
|
| #else |
|
|
| #ifndef HIP_HEADERS__ |
| #include <hip/hip_runtime.h> |
| #include <hip/hip_fp8.h> |
| #include <hip/hip_fp16.h> |
| #include <rocwmma/rocwmma.hpp> |
| #define HIP_HEADERS__ |
| #endif |
|
|
| namespace wmma = rocwmma; |
|
|
| #define LIB_CALL(call) \ |
| do { \ |
| hipError_t err = call; \ |
| if (err != hipSuccess) { \ |
| abort(); \ |
| } \ |
| } while (0) |
|
|
| #define HOST_TYPE(x) hip##x |
|
|
| #endif |
|
|
|
|
