Skip to content

Add alternate method to apply mask to allow XLA to detect MHA pattern - #27

Open
ashors1 wants to merge 1 commit into
google:mainfrom
nvjax-svc-0:patch/alternate-masking-method
Open

Add alternate method to apply mask to allow XLA to detect MHA pattern#27
ashors1 wants to merge 1 commit into
google:mainfrom
nvjax-svc-0:patch/alternate-masking-method

Commits

Commits on Sep 19, 2023