tilelang.rocm.target ==================== .. py:module:: tilelang.rocm.target Attributes ---------- .. autoapisummary:: tilelang.rocm.target.ROCM_MTRIPLE Functions --------- .. autoapisummary:: tilelang.rocm.target.normalize_rocm_arch tilelang.rocm.target.target_get_mcpu tilelang.rocm.target.rocm_warp_size_for_arch tilelang.rocm.target.with_rocm_target_attrs tilelang.rocm.target.check_hip_availability tilelang.rocm.target.normalize_rocm_target tilelang.rocm.target.target_is_hip tilelang.rocm.target.target_is_cdna tilelang.rocm.target.target_is_rdna tilelang.rocm.target.target_is_gfx950 tilelang.rocm.target.target_get_warp_size tilelang.rocm.target.target_get_rdna_generation Module Contents --------------- .. py:data:: ROCM_MTRIPLE :value: 'amdgcn-amd-amdhsa-hcc' .. py:function:: normalize_rocm_arch(arch) .. py:function:: target_get_mcpu(target) .. py:function:: rocm_warp_size_for_arch(arch) .. py:function:: with_rocm_target_attrs(target) .. py:function:: check_hip_availability() Check if HIP (ROCm) is available on the system by locating the ROCm path. :returns: True if HIP is available, False otherwise. :rtype: bool .. py:function:: normalize_rocm_target(target) .. py:function:: target_is_hip(target) .. py:function:: target_is_cdna(target) .. py:function:: target_is_rdna(target) .. py:function:: target_is_gfx950(target) .. py:function:: target_get_warp_size(target) .. py:function:: target_get_rdna_generation(target)