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