Hipify cooperative_groups headers (#3323)

Co-authored-by: N Logan Adams <114770087+loadams@users.noreply.github.com>

Hipify cooperative_groups headers (#3323)
Co-authored-by: N Logan Adams <114770087+loadams@users.noreply.github.com>
6e1cbebe · Ramya Ramineni · GitHub · 4a3ca4e2 · 6e1cbebe · 6e1cbebe
隐藏空白更改
内联并排

Showing with 10 addition and 0 deletion

csrc/includes/quantizer.h csrc/includes/quantizer.h +5 -0

csrc/spatial/includes/spatial_cuda_layers.h csrc/spatial/includes/spatial_cuda_layers.h +5 -0

未找到文件。
--- a/csrc/includes/quantizer.h
+++ b/csrc/includes/quantizer.h
@@ -5,7 +5,12 @@

 #pragma once

+#ifdef __HIP_PLATFORM_HCC__
+#include <hip/hip_cooperative_groups.h>
+#else
 #include <cooperative_groups.h>
+#endif
+
 #include <cuda.h>
 #include <cuda_fp16.h>
 #include <stdio.h>

--- a/csrc/spatial/includes/spatial_cuda_layers.h
+++ b/csrc/spatial/includes/spatial_cuda_layers.h
@@ -9,7 +9,12 @@
 #define HALF_PRECISION_AVAILABLE = 1
 #endif

+#ifdef __HIP_PLATFORM_HCC__
+#include <hip/hip_cooperative_groups.h>
+#else
 #include <cooperative_groups.h>
+#endif
+
 #include <cuda.h>
 #include <cuda_fp16.h>