HIP/MUSA: fix build for backend sampling
This commit is contained in:
parent
804e7e3795
commit
42cf5c01e5
|
|
@ -2,8 +2,12 @@
|
||||||
#include "ggml.h"
|
#include "ggml.h"
|
||||||
#include "softmax.cuh"
|
#include "softmax.cuh"
|
||||||
|
|
||||||
|
#ifdef GGML_USE_HIP
|
||||||
|
#include <hip/hip_cooperative_groups.h>
|
||||||
|
#else
|
||||||
#include <cooperative_groups.h>
|
#include <cooperative_groups.h>
|
||||||
#include <cooperative_groups/reduce.h>
|
#include <cooperative_groups/reduce.h>
|
||||||
|
#endif // GGML_USE_HIP
|
||||||
|
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
|
||||||
|
|
@ -45,6 +45,7 @@
|
||||||
#define cudaDeviceCanAccessPeer hipDeviceCanAccessPeer
|
#define cudaDeviceCanAccessPeer hipDeviceCanAccessPeer
|
||||||
#define cudaDeviceDisablePeerAccess hipDeviceDisablePeerAccess
|
#define cudaDeviceDisablePeerAccess hipDeviceDisablePeerAccess
|
||||||
#define cudaDeviceEnablePeerAccess hipDeviceEnablePeerAccess
|
#define cudaDeviceEnablePeerAccess hipDeviceEnablePeerAccess
|
||||||
|
#define cudaDeviceGetAttribute hipDeviceGetAttribute
|
||||||
#define cudaDeviceProp hipDeviceProp_t
|
#define cudaDeviceProp hipDeviceProp_t
|
||||||
#define cudaDeviceSynchronize hipDeviceSynchronize
|
#define cudaDeviceSynchronize hipDeviceSynchronize
|
||||||
#define cudaError_t hipError_t
|
#define cudaError_t hipError_t
|
||||||
|
|
|
||||||
|
|
@ -56,6 +56,7 @@
|
||||||
#define cudaHostRegisterPortable musaHostRegisterPortable
|
#define cudaHostRegisterPortable musaHostRegisterPortable
|
||||||
#define cudaHostRegisterReadOnly musaHostRegisterReadOnly
|
#define cudaHostRegisterReadOnly musaHostRegisterReadOnly
|
||||||
#define cudaHostUnregister musaHostUnregister
|
#define cudaHostUnregister musaHostUnregister
|
||||||
|
#define cudaLaunchCooperativeKernel musaLaunchCooperativeKernel
|
||||||
#define cudaLaunchHostFunc musaLaunchHostFunc
|
#define cudaLaunchHostFunc musaLaunchHostFunc
|
||||||
#define cudaMalloc musaMalloc
|
#define cudaMalloc musaMalloc
|
||||||
#define cudaMallocHost musaMallocHost
|
#define cudaMallocHost musaMallocHost
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue