Kernels:

kernels-community
/

gemm

Trusted publisher

Kernel card Files Files and versions

gemm / include /gpu_libs.h

drbh

Migrated from kernels-community/gemm

ce83b07 unverified 6 months ago

Raw History Blame Contribute Delete

1.86 kB

	#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