fix array_interface.h half type

2023-05-02 20:50:50 +02:00
parent 65097212b3
commit b324d51f14
3 changed files with 3 additions and 3 deletions
--- a/src/data/array_interface.h
+++ b/src/data/array_interface.h
@@ -603,7 +603,7 @@ void DispatchDType(ArrayInterface<D> const array, std::int32_t device, Fn fn) {
  };
  switch (array.type) {
    case ArrayInterfaceHandler::kF2: {
-#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 600
+#if (defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 600) || defined(__HIP_PLATFORM_AMD__)
      dispatch(__half{});
 #endif
      break;