зеркало из https://github.com/microsoft/DeepSpeed.git
Added rocblas header (#4538)
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
This commit is contained in:
Родитель
beed962c25
Коммит
3e4a587135
|
@ -9,6 +9,9 @@
|
|||
#ifndef __HIP_PLATFORM_HCC__
|
||||
#include <cuda_profiler_api.h>
|
||||
#endif
|
||||
#ifdef __HIP_PLATFORM_HCC__
|
||||
#include <rocblas/rocblas.h>
|
||||
#endif
|
||||
#include <array>
|
||||
#include <cstdio>
|
||||
#include <cstdlib>
|
||||
|
|
Загрузка…
Ссылка в новой задаче