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.

Sign up for GitHub

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

Jump to bottom

feature: DTVB with Swizzling (tensorB) #1562

Open
wants to merge 3 commits into
base: develop
Choose a base branch
from
Open

feature: DTVB with Swizzling (tensorB) #1562

wants to merge 3 commits into from
+226 −51

Conversation

solaslin
Copy link
Contributor

@solaslin solaslin commented Jan 16, 2025 •
edited
Loading

Resolved SWDEV-509997

  • Code-gen of Swizzled (SwizzleTensorB): Requires TN/NN and DTVB
  • FP16, BF16 MFMA_16x16x16_x1. Supported useBias and SAV
  • Included supporting for edge tile and tail-loop
  • Implemented for SwizzledB when WaveGroups[0] > 1 (M-Dim)
  • Support arbitrary M & K for SwizzledB
  • pytests for SwizzledB

[gw1] [ 50%] PASSED Tensile/Tests/common/test_config.py::test_config[Tensile/Tests/self_test/swizzleB.yaml]
[gw0] [100%] PASSED Tensile/Tests/common/test_config.py::test_config[Tensile/Tests/self_test/dtl.yaml]
=== 2 passed, 5 warnings in 449.50s (0:07:29) ===
py310: OK (592.75=setup[10.28]+cmd[0.58,132.13,449.76] seconds)
congratulations :) (592.86 seconds)

@solaslin solaslin added the noCI Disable testing on supported CI systems: math libraries CI has this feature enabled.. label Jan 16, 2025
@solaslin solaslin self-assigned this Jan 16, 2025
@solaslin solaslin force-pushed the swizzledB branch 3 times, most recently from 0d524ab to 5bc3ae7 Compare January 21, 2025 06:43
@solaslin solaslin added enhancement New feature or request and removed noCI Disable testing on supported CI systems: math libraries CI has this feature enabled.. labels Jan 23, 2025
@solaslin solaslin marked this pull request as ready for review January 23, 2025 03:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@jichangjichang jichangjichang Awaiting requested review from jichangjichang jichangjichang is a code owner

@KKyang KKyang Awaiting requested review from KKyang KKyang is a code owner

@vin-huang vin-huang Awaiting requested review from vin-huang vin-huang is a code owner

@imcarsonliao imcarsonliao Awaiting requested review from imcarsonliao imcarsonliao is a code owner

@hcman2 hcman2 Awaiting requested review from hcman2 hcman2 is a code owner

@Serge45 Serge45 Awaiting requested review from Serge45 Serge45 is a code owner

@Jinp800125 Jinp800125 Awaiting requested review from Jinp800125 Jinp800125 is a code owner

@TonyYHsieh TonyYHsieh Awaiting requested review from TonyYHsieh TonyYHsieh is a code owner

At least 2 approving reviews are required to merge this pull request.

Assignees

@solaslin solaslin

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@solaslin