Skip to content

Remove z-shape block swizzle#92

Merged
zasdfgbnm merged 1 commit intomainfrom
zshape
Mar 29, 2023
Merged

Remove z-shape block swizzle#92
zasdfgbnm merged 1 commit intomainfrom
zshape

Conversation

@zasdfgbnm
Copy link
Collaborator

z-shape swizzle was not used in tracking-matmul, and now we have a better swizzle implemented in #90

// CTA tile:

// Swizzle block tiles:
c->swizzle(Swizzle2DType::ZShape, 0, 1, SwizzleMode::Loop);
Copy link
Contributor

@drzejan2 drzejan2 Mar 29, 2023

Choose a reason for hiding this comment

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

It looks like this is the only place where TensorView::swizzle function is currently used in production.

Do you think it should be kept, or should it be removed, along with tests for it? (test_gpu_tensorcore.cpp and test_gpu_swizzle.cpp). Perhaps there are use cases where it will be required, but nothing comes to my mind right away.

Other than this, PR looks good to me.

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

TensorView::swizzle should also be applied to the shared memory buffer to remove bank conflict. It is now in main branch now, but #41 will add it.

@zasdfgbnm zasdfgbnm merged commit 5730a19 into main Mar 29, 2023
@zasdfgbnm zasdfgbnm deleted the zshape branch March 29, 2023 20:01
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.

2 participants