Code

[ROCm] Add GemmFloat8#18488

Code

Merged

cloudhan merged 14 commits intomainmicrosoft/onnxruntime:mainfrom guangyunhan/rocm-fp8gemmmicrosoft/onnxruntime:guangyunhan/rocm-fp8gemmCopy head branch name to clipboardDec 11, 2023

+1,648-36

Commits

Commits on Dec 5, 2023

• Add Op GemmFloat8 and implement kernels with ck

  cloudhan

  committed
  c05fc49

• Add transb support for fp8 b

  cloudhan

  committed
  508643d

• Lint

  cloudhan

  committed
  c315fd5

• Update test to allow adding ROCm EP

  cloudhan

  committed
  97ab0da

• Format

  cloudhan

  committed
  6b0bdd0

• Add GemmFloat8 OpKernel test

  cloudhan

  committed
  9d10672

• Add generic

  cloudhan

  committed
  9dbeb80

• Minor

  cloudhan

  committed
  0e4649a

• Fix ifdef

  cloudhan

  committed
  f12c2f2

• Add ml_dtypes for testing

  cloudhan

  committed
  da55d2c

• Minor

  cloudhan

  committed
  389aba9

• Fix mnk

  cloudhan

  committed
  eeeccfd

• Rename ke binding

  cloudhan

  committed
  5b6deb5

• Add tunable binding

  cloudhan

  committed
  c564bae