Add tests for ck mha fp16 solver

ROCm / MIOpen

AMD's Machine Intelligence Library

https://rocm.docs.amd.com/projects/MIOpen/en/latest/

Other

1.09k stars 230 forks source link

Add tests for ck mha fp16 solver #3304

Closed BrianHarrisonAMD closed 1 month ago

BrianHarrisonAMD commented 1 month ago

Purpose of review:

Add tests for CK FA V2 FWD inference solver.
Follow up to #3277

Note:

Merging these changes into the integration branch to allow for limited diff, but will follow up with a PR for merging the integration branch into develop once complete.
Testing and solver are limited to FWD inference for now.

BrianHarrisonAMD commented 1 month ago

Note #3302 updates requirement.txt CK hash to the version needed for the integration branch. Ill wait until #3302 is merged to develop before merging the integration branch to develop in order to align these properly (it'll probably be merged first anyways, but mentioning just in case).