remove unless include

tocean · wenchenvincent · commit 11b799d8cb58 · 2024-01-30T07:09:09.000Z
diff --git a/csrc/dist_op/dist.cpp b/csrc/dist_op/dist.cpp
@@ -4,7 +4,7 @@
 #include <stdio.h>
 #include <dlfcn.h>
 
-#if defined(__HIP_PLATFORM_AMD__)
+#ifdef __HIP_PLATFORM_AMD__
 #include <hip/hip_runtime.h>
 #include <rccl.h>
 #define stream_t hipStream_t
diff --git a/csrc/extensions/include/common.h b/csrc/extensions/include/common.h
@@ -4,8 +4,6 @@
 #ifndef MSAMP_COMMON_H_
 #define MSAMP_COMMON_H_
 
-#include <cublasLt.h>
-#include <cublas_v2.h>
 #include <cuda_bf16.h>
 #include <cuda_fp16.h>
 #include <cuda_fp8.h>
diff --git a/csrc/extensions/include/utils.cuh b/csrc/extensions/include/utils.cuh
@@ -9,8 +9,6 @@
 #ifndef MSAMP_UTILS_CUH_
 #define MSAMP_UTILS_CUH_
 
-#include <cuda_bf16.h>
-#include <cuda_fp16.h>
 #include <cstdint>
 #include <cassert>
 
diff --git a/csrc/extensions/optim/adamw.cu b/csrc/extensions/optim/adamw.cu
@@ -2,8 +2,6 @@
 // Licensed under the MIT License.
 
 #include <c10/cuda/CUDAGuard.h>
-#include <cublasLt.h>
-#include <cublas_v2.h>
 #include <torch/extension.h>
 
 #include <cmath>
