Commit fd32631

authored and

committed

Migrate GenAI kv cache kernel arguments to use PTA_B (#4887)

Summary: Pull Request resolved: #4887 - Migrate GenAI kv cache kernel arguments to use `PTA_B` Reviewed By: r-barnes Differential Revision: D81703942 fbshipit-source-id: 0577756dd30fa669477270d5c42a8057ec4f2935

1 parent 57c2293 commit fd32631Copy full SHA for fd32631

1 file changed

+95

-140

lines changed

fbgemm_gpu/experimental/gen_ai/src/kv_cache
- kv_cache.cu

1 file changed

+95

-140

lines changed

Comments

(0)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit fd32631

1 file changed

1 file changed

File tree

1 file changed

1 file changed

0 commit comments