NVIDIA / NVIDIA/cccl

Use `BlockLoadToShared` in transpose versions of `BlockLoad`

Open
#9,057 6 comments 0 reactions 1 assignee Claimed by @pauleonix View on GitHub
Dominant language
C++
Stars
2.5k
Forks
486
Avg merge
2d 6h
Merged PRs (30d)
295

Description

All transposing versions of `cub::BlockLoad` load data from GMEM to registers, then store those back into SMEM and load them again in a different order to registers. Using `BlockLoadToShared` would save the round trip through SMEM. We should try if `BlockLoadToShared` can improve the performance of at least one transforming use of block load in one CUB algorithm to validate this hypothesis.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.