Add include for fp16 on HIP

trackreco · Aug 12, 2024 · 22192ec · 22192ec
1 parent a0432cc
commit 22192ec
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/RecoTracker/LSTCore/interface/alpaka/Constants.h b/RecoTracker/LSTCore/interface/alpaka/Constants.h
@@ -3,8 +3,10 @@
 
 #include "RecoTracker/LSTCore/interface/Constants.h"
 
-#ifdef ALPAKA_ACC_GPU_CUDA_ENABLED
+#if defined ALPAKA_ACC_GPU_CUDA_ENABLED
 #include <cuda_fp16.h>
+#elif defined ALPAKA_ACC_GPU_HIP_ENABLED
+#include <hip/hip_fp16.h>
 #endif
 
 namespace lst {