mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-29 11:16:59 +00:00
35de33c57b
The following examples have been tested for 04_codegen: ./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 256 256 ./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 64 64 ./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 32 32 ./bin/codegen_basic_flash_attention_fwd 1 1 64 4096 4096 128 128 ./bin/codegen_basic_flash_attention_fwd 1 1 64 2048 2048 128 128 ./bin/codegen_basic_flash_attention_fwd 1 1 64 512 512 128 128