[DCU] Preliminary adaptation

transformer_engine/pytorch/cpp_extensions/__init__.py

@@ -5,5 +5,7 @@
 """Python interface for c++ extensions"""
 from transformer_engine_torch import *
 
-from .fused_attn import *
+from torch.utils.cpp_extension import IS_HIP_EXTENSION
+if not IS_HIP_EXTENSION:
+  from .fused_attn import *
 from .gemm import *