Fix build error

Signed-off-by: wenjh <wenjh@sugon.com>

transformer_engine/common/swizzle/swizzle_block_scaling.cu

@@ -14,6 +14,7 @@
 #include "../util/logging.h"
 #include "transformer_engine/transformer_engine.h"
 
+#ifdef __HIP_PLATFORM_AMD__
 namespace transformer_engine {
 namespace {
 constexpr uint32_t WARP_SIZE = 32;
@@ -311,11 +312,16 @@ void swizzle_block_scaling_to_mxfp8_scaling_factors(const Tensor* input, Tensor*
 }
 
 }  // namespace transformer_engine
+#endif
 
 void nvte_swizzle_block_scaling_to_mxfp8_scaling_factors(const NVTETensor input, NVTETensor output,
                                                          cudaStream_t stream) {
   NVTE_API_CALL(nvte_swizzle_block_scaling_to_mxfp8_scaling_factors);
+#ifdef __HIP_PLATFORM_AMD__
+  NVTE_CHECK(false, "nvte_swizzle_block_scaling_to_mxfp8_scaling_factors is not supported on rocm");
+#else
   using namespace transformer_engine;
   swizzle_block_scaling_to_mxfp8_scaling_factors(convertNVTETensorCheck(input),
                                                  convertNVTETensorCheck(output), stream);
+#endif
 }