Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[WIP] TMA Version of HSTU (Autotuned) #71

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

plotfi
Copy link

@plotfi plotfi commented Aug 21, 2024

Based on #57, this version uses the autotuned to toggle use of TMA.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Aug 21, 2024
@plotfi
Copy link
Author

plotfi commented Aug 21, 2024

@manman-ren @embg

ops/triton/triton_ragged_hstu_attention.py Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Outdated Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Outdated Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Outdated Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Outdated Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Outdated Show resolved Hide resolved
ops/triton/triton_ragged_hstu_attention.py Outdated Show resolved Hide resolved
@plotfi plotfi force-pushed the plotfi-HSTU-TMA-autotuned branch 5 times, most recently from d06298d to 0446bb0 Compare September 4, 2024 02:06
@plotfi
Copy link
Author

plotfi commented Sep 6, 2024

@manman-ren Updated, let me know what you think. Will try and get this running with the OSS benchmark launcher.

@manman-ren
Copy link

looks good! Thanks

Based on facebookresearch#57, this version uses the autotuned to toggle use of TMA.
@plotfi
Copy link
Author

plotfi commented Sep 7, 2024

looks good! Thanks

went ahead and cleaned up the autotuning setup. I also got a test launcher running but I am not sure if it is doing things correctly

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants