Repository navigation

Commit 8d4bf64

committed

Feat: internal 1d and 2d ring attention kernel

1 parent 8e3e843 commit 8d4bf64Copy full SHA for 8d4bf64

6 files changed

src/maxdiffusion
- kernels
  - internal_ring_attention.py
- models
  - attention_flax.py
- pyconfig.py
- tests
  - tile_size_grid_search_test.py
- utils
  - tile_size_grid_search.py
  - wan_block_benchmark.py

Comments

(0)