Skip to content

Fix softmax multi-block path for n_cols > 65536 (unreachable + non-compiling)#1252

Open
lollinng wants to merge 2 commits into
linkedin:mainfrom
lollinng:fix/softmax-large-vocab-multiblock
Open

Fix softmax multi-block path for n_cols > 65536 (unreachable + non-compiling)#1252
lollinng wants to merge 2 commits into
linkedin:mainfrom
lollinng:fix/softmax-large-vocab-multiblock

test: exercise softmax multi-block path with n_cols > MAX_FUSED_SIZE

4b9b2c5
Select commit
Loading
Failed to load commit list.