MANIFEST.in
pyproject.toml
setup.py
3rdparty/hipify_torch/.git
3rdparty/hipify_torch/.gitignore
3rdparty/hipify_torch/LICENSE.txt
3rdparty/hipify_torch/README.md
3rdparty/hipify_torch/hipify_cli.py
3rdparty/hipify_torch/setup.py
3rdparty/hipify_torch/cmake/Hipify.cmake
3rdparty/hipify_torch/hipify_torch/__init__.py
3rdparty/hipify_torch/hipify_torch/constants.py
3rdparty/hipify_torch/hipify_torch/cuda_to_hip_mappings.py
3rdparty/hipify_torch/hipify_torch/hipify_python.py
3rdparty/hipify_torch/hipify_torch/version.py
3rdparty/hipify_torch/hipify_torch/__pycache__/__init__.cpython-312.pyc
3rdparty/hipify_torch/hipify_torch/__pycache__/version.cpython-312.pyc
3rdparty/hipify_torch/hipify_torch/v2/__init__.py
3rdparty/hipify_torch/hipify_torch/v2/cuda_to_hip_mappings.py
3rdparty/hipify_torch/hipify_torch/v2/hipify_python.py
3rdparty/hipify_torch/hipify_torch/v2/__pycache__/__init__.cpython-312.pyc
3rdparty/hipify_torch/hipify_torch/v2/__pycache__/cuda_to_hip_mappings.cpython-312.pyc
3rdparty/hipify_torch/hipify_torch/v2/__pycache__/hipify_python.cpython-312.pyc
3rdparty/hipify_torch/test/test_installation.py
3rdparty/hipify_torch/tools/replace_cuda_with_hip_files.py
build_tools/VERSION.txt
build_tools/__init__.py
build_tools/build_ext.py
build_tools/jax.py
build_tools/pytorch.py
build_tools/rocm_utils.cmake
build_tools/te_version.py
build_tools/utils.py
build_tools/__pycache__/__init__.cpython-312.pyc
build_tools/__pycache__/build_ext.cpython-312.pyc
build_tools/__pycache__/pytorch.cpython-312.pyc
build_tools/__pycache__/te_version.cpython-312.pyc
build_tools/__pycache__/utils.cpython-312.pyc
build_tools/hipify/custom_map.json
build_tools/hipify/hipify.cmake
build_tools/hipify/hipify.py
build_tools/hipify/__pycache__/hipify.cpython-312.pyc
build_tools/wheel_utils/Dockerfile.aarch
build_tools/wheel_utils/Dockerfile.rocm.manylinux.x86
build_tools/wheel_utils/Dockerfile.x86
build_tools/wheel_utils/build_wheels.sh
build_tools/wheel_utils/launch_aarch.sh
build_tools/wheel_utils/launch_x86.sh
common_headers/common/common.h
common_headers/common/common_hip.h
common_headers/common/cudnn_utils.h
common_headers/common/cudnn_utils_hip.h
common_headers/common/nvtx.h
common_headers/common/nvtx_hip.h
common_headers/common/utils.cuh
common_headers/common/utils_hip.cuh
common_headers/common/activation/activation_template.h
common_headers/common/activation/activation_template_hip.h
common_headers/common/amd_detail/hip_float8.h
common_headers/common/cast/core/common.cuh
common_headers/common/cast/core/common_hip.cuh
common_headers/common/cast/dispatch/dequantize.cuh
common_headers/common/cast/dispatch/dequantize_hip.cuh
common_headers/common/cast/dispatch/gated.cuh
common_headers/common/cast/dispatch/gated_hip.cuh
common_headers/common/cast/dispatch/quantize.cuh
common_headers/common/cast/dispatch/quantize_hip.cuh
common_headers/common/cast/fp8/dequantize_fp8.cuh
common_headers/common/cast/fp8/dequantize_fp8_hip.cuh
common_headers/common/cast/fp8/gated_fp8.cuh
common_headers/common/cast/fp8/gated_fp8_hip.cuh
common_headers/common/cast/fp8/quantize_fp8.cuh
common_headers/common/cast/fp8/quantize_fp8_hip.cuh
common_headers/common/cast/mxfp8/dequantize_mxfp8.cuh
common_headers/common/cast/mxfp8/dequantize_mxfp8_hip.cuh
common_headers/common/cast/mxfp8/gated_mxfp8.cuh
common_headers/common/cast/mxfp8/gated_mxfp8_hip.cuh
common_headers/common/cast/mxfp8/quantize_mxfp8.cuh
common_headers/common/cast/mxfp8/quantize_mxfp8_hip.cuh
common_headers/common/cast/mxfp8/rocm_dequantize_mxfp8.cuh
common_headers/common/cast/mxfp8/rocm_gated_mxfp8.cuh
common_headers/common/cast/mxfp8/rocm_quantize_mxfp8.cuh
common_headers/common/cast/mxfp8/rocm_vectorized_2d.cuh
common_headers/common/cast/mxfp8/rocm_vectorized_2d_hip.cuh
common_headers/common/cast/nvfp4/core_nvfp4.cuh
common_headers/common/cast/nvfp4/core_nvfp4_hip.cuh
common_headers/common/cast/nvfp4/dequantize_nvfp4.cuh
common_headers/common/cast/nvfp4/dequantize_nvfp4_hip.cuh
common_headers/common/cast/nvfp4/quantize_nvfp4.cuh
common_headers/common/cast/nvfp4/quantize_nvfp4_hip.cuh
common_headers/common/cast/nvfp4/quantize_transpose_nvfp4.cuh
common_headers/common/cast/nvfp4/quantize_transpose_nvfp4_hip.cuh
common_headers/common/comm_gemm_overlap/userbuffers/ipcsocket.h
common_headers/common/comm_gemm_overlap/userbuffers/userbuffers.h
common_headers/common/comm_gemm_overlap/userbuffers/userbuffers_hip.h
common_headers/common/fused_attn/fused_attn_f16_arbitrary_seqlen.h
common_headers/common/fused_attn/fused_attn_f16_arbitrary_seqlen_hip.h
common_headers/common/fused_attn/fused_attn_f16_max512_seqlen.h
common_headers/common/fused_attn/fused_attn_f16_max512_seqlen_hip.h
common_headers/common/fused_attn/fused_attn_fp8.h
common_headers/common/fused_attn/fused_attn_fp8_hip.h
common_headers/common/fused_attn/utils.h
common_headers/common/fused_attn/utils_hip.h
common_headers/common/fused_attn_rocm/fused_attn_aotriton.h
common_headers/common/fused_attn_rocm/fused_attn_aotriton_hip.h
common_headers/common/fused_attn_rocm/fused_attn_ck.h
common_headers/common/fused_attn_rocm/fused_attn_ck_hip.h
common_headers/common/fused_attn_rocm/utils.h
common_headers/common/fused_attn_rocm/utils_hip.h
common_headers/common/fused_router/utils.h
common_headers/common/fused_router/utils_hip.h
common_headers/common/gemm/ck_grouped_gemm.h
common_headers/common/gemm/config.h
common_headers/common/gemm/config_hip.h
common_headers/common/gemm/cutlass_grouped_gemm.cuh
common_headers/common/gemm/cutlass_grouped_gemm_hip.cuh
common_headers/common/include/transformer_engine/activation.h
common_headers/common/include/transformer_engine/activation_hip.h
common_headers/common/include/transformer_engine/cast.h
common_headers/common/include/transformer_engine/cast_hip.h
common_headers/common/include/transformer_engine/cast_transpose_noop.h
common_headers/common/include/transformer_engine/cast_transpose_noop_hip.h
common_headers/common/include/transformer_engine/comm_gemm.h
common_headers/common/include/transformer_engine/comm_gemm_hip.h
common_headers/common/include/transformer_engine/comm_gemm_overlap.h
common_headers/common/include/transformer_engine/comm_gemm_overlap_hip.h
common_headers/common/include/transformer_engine/cudnn.h
common_headers/common/include/transformer_engine/cudnn_hip.h
common_headers/common/include/transformer_engine/dropout.h
common_headers/common/include/transformer_engine/dropout_hip.h
common_headers/common/include/transformer_engine/fused_attn.h
common_headers/common/include/transformer_engine/fused_attn_hip.h
common_headers/common/include/transformer_engine/fused_rope.h
common_headers/common/include/transformer_engine/fused_rope_hip.h
common_headers/common/include/transformer_engine/fused_router.h
common_headers/common/include/transformer_engine/fused_router_hip.h
common_headers/common/include/transformer_engine/gemm.h
common_headers/common/include/transformer_engine/gemm_hip.h
common_headers/common/include/transformer_engine/hadamard_transform.h
common_headers/common/include/transformer_engine/hadamard_transform_hip.h
common_headers/common/include/transformer_engine/multi_stream.h
common_headers/common/include/transformer_engine/multi_stream_hip.h
common_headers/common/include/transformer_engine/multi_tensor.h
common_headers/common/include/transformer_engine/multi_tensor_hip.h
common_headers/common/include/transformer_engine/normalization.h
common_headers/common/include/transformer_engine/normalization_hip.h
common_headers/common/include/transformer_engine/padding.h
common_headers/common/include/transformer_engine/padding_hip.h
common_headers/common/include/transformer_engine/permutation.h
common_headers/common/include/transformer_engine/permutation_hip.h
common_headers/common/include/transformer_engine/recipe.h
common_headers/common/include/transformer_engine/recipe_hip.h
common_headers/common/include/transformer_engine/softmax.h
common_headers/common/include/transformer_engine/softmax_hip.h
common_headers/common/include/transformer_engine/swizzle.h
common_headers/common/include/transformer_engine/swizzle_hip.h
common_headers/common/include/transformer_engine/transformer_engine.h
common_headers/common/include/transformer_engine/transformer_engine_hip.h
common_headers/common/include/transformer_engine/transpose.h
common_headers/common/include/transformer_engine/transpose_hip.h
common_headers/common/multi_tensor/multi_tensor_apply.cuh
common_headers/common/multi_tensor/multi_tensor_apply_hip.cuh
common_headers/common/normalization/common.h
common_headers/common/normalization/common_hip.h
common_headers/common/normalization/kernel_traits.h
common_headers/common/normalization/kernel_traits_hip.h
common_headers/common/normalization/layernorm/ln_bwd_kernels.cuh
common_headers/common/normalization/layernorm/ln_bwd_kernels_hip.cuh
common_headers/common/normalization/layernorm/ln_fwd_kernels.cuh
common_headers/common/normalization/layernorm/ln_fwd_kernels_hip.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_bwd_kernels.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_bwd_kernels_hip.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_fwd_kernels.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_fwd_kernels_hip.cuh
common_headers/common/nvshmem_api/nvshmem_waitkernel.h
common_headers/common/nvshmem_api/nvshmem_waitkernel_hip.h
common_headers/common/recipe/recipe_common.cuh
common_headers/common/recipe/recipe_common_hip.cuh
common_headers/common/transpose/cast_transpose.h
common_headers/common/transpose/cast_transpose_hip.h
common_headers/common/util/cuda_driver.h
common_headers/common/util/cuda_nvml.h
common_headers/common/util/cuda_runtime.h
common_headers/common/util/handle_manager.h
common_headers/common/util/handle_manager_hip.h
common_headers/common/util/hip_driver.h
common_headers/common/util/hip_nvml.h
common_headers/common/util/hip_runtime.h
common_headers/common/util/logging.h
common_headers/common/util/logging_hip.h
common_headers/common/util/math.h
common_headers/common/util/multi_stream.h
common_headers/common/util/multi_stream_hip.h
common_headers/common/util/ptx.cuh
common_headers/common/util/ptx_hip.cuh
common_headers/common/util/pybind_helper.h
common_headers/common/util/pybind_helper_hip.h
common_headers/common/util/rtc.h
common_headers/common/util/rtc_hip.h
common_headers/common/util/shared_lib_wrapper.h
common_headers/common/util/string.h
common_headers/common/util/system.h
common_headers/common/util/system_hip.h
common_headers/common/util/vectorized_pointwise.h
common_headers/common/util/vectorized_pointwise_hip.h
csrc/common.cpp
csrc/common.h
csrc/extensions.h
csrc/pybind.h
csrc/quantizer.cpp
csrc/type_converters.cpp
csrc/util.cpp
csrc/util.h
csrc/extensions/activation.cpp
csrc/extensions/apply_rope.cpp
csrc/extensions/attention.cpp
csrc/extensions/bias.cpp
csrc/extensions/cast.cpp
csrc/extensions/comm_gemm_overlap.cpp
csrc/extensions/dropout.cpp
csrc/extensions/fp8_block_scaling_partial_cast.cpp
csrc/extensions/gemm.cpp
csrc/extensions/misc.cpp
csrc/extensions/normalization.cpp
csrc/extensions/nvshmem_comm.cpp
csrc/extensions/padding.cpp
csrc/extensions/permutation.cpp
csrc/extensions/pybind.cpp
csrc/extensions/recipe.cpp
csrc/extensions/rocshmem_comm.cpp
csrc/extensions/router.cpp
csrc/extensions/softmax.cpp
csrc/extensions/transpose.cpp
csrc/extensions/multi_tensor/adam.cpp
csrc/extensions/multi_tensor/compute_scale.cpp
csrc/extensions/multi_tensor/l2norm.cpp
csrc/extensions/multi_tensor/scale.cpp
csrc/extensions/multi_tensor/sgd.cpp
transformer_engine_rocm_torch.egg-info/PKG-INFO
transformer_engine_rocm_torch.egg-info/SOURCES.txt
transformer_engine_rocm_torch.egg-info/dependency_links.txt
transformer_engine_rocm_torch.egg-info/requires.txt
transformer_engine_rocm_torch.egg-info/top_level.txt
triton_kernels/gmm/configs/gfx942-GMM.json
triton_kernels/gmm/configs/gfx950-GMM.json