Skip to content

[Epilogue] Align fixed visitor workspace offsets - #3631

Open
XFDG wants to merge 1 commit into
NVIDIA:mainfrom
XFDG:fix/sm90-visitor-workspace-alignment-3494
Open

XFDG wants to merge 1 commit into
NVIDIA:mainfrom
XFDG:fix/sm90-visitor-workspace-alignment-3494

Conversation

@XFDG

@XFDG XFDG commented Sep 13, 2026

Copy link
Copy Markdown

Summary

  • align fixed-size Sm90VisitorImplBase workspace pointer advances with MinWorkspaceAlignment
  • keep pointer offsets consistent with get_workspace_size() and initialize_workspace()
  • add coverage for the 2-, 3-, and 4-operation specializations

Testing

Tested on NVIDIA B200 (SM100) with CUDA 13.1:

  • baseline reproducer: offsets 0,3,8,15 (failed as expected)
  • fixed reproducer: offsets 0,16,32,48 (passed)
  • cutlass_test_unit_epilogue_threadblock --gtest_filter=Epilogue_Threadblock.Sm90VisitorWorkspaceAlignment (1/1 passed)

Contribution disclosure

This contribution was developed with assistance from OpenAI Codex.

Fixes #3494

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.

Sm90VisitorImplBase fixed-size specializations advance workspace by raw sizes while get_workspace_size/initialize_workspace round to 16

1 participant