hip : Add hipGraph and VMM support to ROCM (#11362)
* Add hipGraph support * Enable VMM on rocm
This commit is contained in:
parent
c5d9effb49
commit
5f0db9522f
5 changed files with 92 additions and 20 deletions
|
@ -588,7 +588,7 @@ struct ggml_tensor_extra_gpu {
|
|||
};
|
||||
|
||||
|
||||
#if (CUDART_VERSION >= 12000) && defined(GGML_CUDA_USE_GRAPHS)
|
||||
#if ((CUDART_VERSION >= 12000) && defined(GGML_CUDA_USE_GRAPHS)) || defined(GGML_HIP_GRAPHS)
|
||||
#define USE_CUDA_GRAPH
|
||||
#endif
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue