Skip to content

[PyTorch] Compile-friendly attention API: pointer-free qkv layout detection - #16

Open
pggPL wants to merge 2 commits into
mainfrom
compile_friendly_attention_api
Open

[PyTorch] Compile-friendly attention API: pointer-free qkv layout detection#16
pggPL wants to merge 2 commits into
mainfrom
compile_friendly_attention_api

[PyTorch] attention: route traced get_qkv_layout calls to the pointer…

475f292
Select commit
Loading
Failed to load commit list.