Skip to content

[ROCm] Facilitate shfl_xor usage in AMD platform#3664

Closed
yiakwy-xpu-ml-framework-team wants to merge 5 commits intosgl-project:use__shfl_xor_sync_style_reducefrom
yiakwy-xpu-ml-framework-team:enable_shfl_xor_sync_style_reduce_on_amd
Closed

[ROCm] Facilitate shfl_xor usage in AMD platform#3664
yiakwy-xpu-ml-framework-team wants to merge 5 commits intosgl-project:use__shfl_xor_sync_style_reducefrom
yiakwy-xpu-ml-framework-team:enable_shfl_xor_sync_style_reduce_on_amd

Conversation

@yiakwy-xpu-ml-framework-team
Copy link
Copy Markdown
Contributor

@yiakwy-xpu-ml-framework-team yiakwy-xpu-ml-framework-team commented Feb 18, 2025

Motivation

Enabe compiling in AMD platform :

SDK : ROCM 6.3
CHIP ARCH : MI series

Modifications

Checklist

Comment thread sgl-kernel/src/sgl-kernel/csrc/per_token_group_quant_fp8.cu Outdated
Comment thread sgl-kernel/src/sgl-kernel/include/utils.h
Comment thread sgl-kernel/src/sgl-kernel/include/hip_math_def.h Outdated
#ifndef USE_ROCM
#ifdef USE_ROCM

#define CHECK_INPUT(x) \
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Aten has already defined CHECK_INPUT?

Comment thread sgl-kernel/src/sgl-kernel/ops/__init__.py Outdated
Comment thread sgl-kernel/src/sgl-kernel/include/utils.h
@yiakwy-xpu-ml-framework-team yiakwy-xpu-ml-framework-team changed the title facilitate compiling in AMD platform [ROCm] facilitate compiling in AMD platform Feb 19, 2025
@yiakwy-xpu-ml-framework-team yiakwy-xpu-ml-framework-team changed the title [ROCm] facilitate compiling in AMD platform [ROCm] Facilitate shlf_xor usage in AMD platform Feb 19, 2025
@yiakwy-xpu-ml-framework-team yiakwy-xpu-ml-framework-team changed the title [ROCm] Facilitate shlf_xor usage in AMD platform [ROCm] Facilitate shfl_xor usage in AMD platform Feb 19, 2025
@zhyncs zhyncs deleted the branch sgl-project:use__shfl_xor_sync_style_reduce February 19, 2025 12:53
@zhyncs zhyncs closed this Feb 19, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants