Fp8 hipstream fix (#3127)

Summary: X-link: facebookresearch/FBGEMM#333 Pick hip stream where cuda is not defined Pull Request resolved: #3127 Reviewed By: jiawenliu64 Differential Revision: D63662576 Pulled By: xw285cornell fbshipit-source-id: cacecdd2401a5fda0cec05831a04ecd40edd4e6d
pytorch · Oct 10, 2024 · 8e7beba · 8e7beba
1 parent f14ca4d
commit 8e7beba
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 16 deletions.
diff --git a/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_blockwise_gemm.hip b/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_blockwise_gemm.hip
@@ -15,8 +15,6 @@
 #include <c10/hip/HIPStream.h>
 #include <torch/torch.h>
 
-#if defined(USE_ROCM)
-
 #include "ck/ck.hpp"
 #include "ck/tensor_operation/gpu/device/gemm_specialization.hpp"
 #include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
@@ -272,5 +270,3 @@ at::Tensor f8f8bf16_blockwise(
 }
 
 } // namespace fbgemm_gpu
-
-#endif // defined(USE_ROCM)
diff --git a/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_rowwise_gemm.hip b/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_rowwise_gemm.hip
@@ -18,8 +18,6 @@
 #include <c10/hip/HIPStream.h>
 #include <torch/torch.h>
 
-#if defined(USE_ROCM)
-
 #include "kernels/fp8_rowwise_kernel_manifest.h"
 
 namespace fbgemm_gpu {
@@ -302,5 +300,3 @@ at::Tensor f8f8bf16_rowwise(
 }
 
 } // namespace fbgemm_gpu
-
-#endif // defined(USE_ROCM)
diff --git a/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_tensorwise_gemm.hip b/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_tensorwise_gemm.hip
@@ -15,8 +15,6 @@
 #include <c10/hip/HIPStream.h>
 #include <torch/torch.h>
 
-#if defined(USE_ROCM)
-
 #include "ck/ck.hpp"
 #include "ck/tensor_operation/gpu/device/gemm_specialization.hpp"
 #include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
@@ -243,5 +241,3 @@ at::Tensor f8f8bf16_tensorwise(
 }
 
 } // namespace fbgemm_gpu
-
-#endif // defined(USE_ROCM)
diff --git a/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/kernels/fp8_rowwise_common.h b/fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/kernels/fp8_rowwise_common.h
@@ -15,8 +15,6 @@
 #include <c10/cuda/CUDAStream.h>
 #include <torch/torch.h>
 
-#if defined(USE_ROCM)
-
 #include "ck/ck.hpp"
 #include "ck/tensor_operation/gpu/device/gemm_specialization.hpp"
 #include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
@@ -200,5 +198,3 @@ at::Tensor f8f8bf16_rowwise_impl(
 
   return Y;
 }
-
-#endif // defined(USE_ROCM)