Code

Make CUDA a NHWC EP#17200

Code

Merged

hariharans29 merged 25 commits intomicrosoft:mainmicrosoft/onnxruntime:mainfrom gedoensmax:cuda_nhwcgedoensmax/onnxruntime:cuda_nhwcCopy head branch name to clipboardOct 16, 2023

+1,198-287

Commits

Commits on Oct 9, 2023

• benchmarking utilities

  gedoensmax

  committed
  cbecb0d

  Copy full SHA for cbecb0d

• option to make CUDA EP prefer NHWC

  gedoensmax

  committed
  8917c74

  Copy full SHA for 8917c74

• register NHWC conv

  gedoensmax

  committed
  2cdd184

  Copy full SHA for 2cdd184

• batch- and instance norm NHWC

  gedoensmax

  committed
  51db617

  Copy full SHA for 51db617

• poling NHWC op

  gedoensmax

  committed
  5e0e235

  Copy full SHA for 5e0e235

• resolving hacky NHWC command line parsing and enabling arbitraty options

  gedoensmax

  committed
  3ab4f85

  Copy full SHA for 3ab4f85

• add copyright header

  gedoensmax

  committed
  fba5864

  Copy full SHA for fba5864

• CUDA NHWC unit test

  gedoensmax

  committed
  5a258ec

  Copy full SHA for 5a258ec

• using pre pack to transpose weights

  gedoensmax

  committed
  ea0d033

  Copy full SHA for ea0d033

• Adding more unit tests

  gedoensmax

  committed
  0aa7e51

  Copy full SHA for 0aa7e51

• remove becnh script

  gedoensmax

  committed
  7d040a9

  Copy full SHA for 7d040a9

• dropping instance norm and moving to typed tests

  gedoensmax

  committed
  f3625c3

  Copy full SHA for f3625c3

• adress review comments

  gedoensmax

  committed
  7b808a3

  Copy full SHA for 7b808a3

• adding compile option to omit NHWC kernels

  gedoensmax

  committed
  7892929

  Copy full SHA for 7892929

• remove remaining fused conv and filter NHWC supported ops

  gedoensmax

  committed
  fff1860

  Copy full SHA for fff1860

• review change cleanup

  gedoensmax

  committed
  d767a6d

  Copy full SHA for d767a6d

• adding pre pack sync

  gedoensmax

  committed
  9df8fe2

  Copy full SHA for 9df8fe2

• lint changes

  gedoensmax

  committed
  8d12c11

  Copy full SHA for 8d12c11

Commits on Oct 10, 2023

• support contrib NHWC conv

  gedoensmax

  committed
  3ad8e5c

  Copy full SHA for 3ad8e5c

• update include

  Show description for 43cc600
  

  gedoensmax

  and

  tianleiwu

  authored
  43cc600

  Copy full SHA for 43cc600

• revert documentation of prepack bool

  gedoensmax

  committed
  8b15fb8

  Copy full SHA for 8b15fb8

Commits on Oct 11, 2023

• fromatting and adding more comments

  gedoensmax

  committed
  b2402a9

  Copy full SHA for b2402a9

Commits on Oct 12, 2023

• cpplint changes

  gedoensmax

  committed
  d0daf4e

  Copy full SHA for d0daf4e

• fix error messages

  gedoensmax

  committed
  1cd4672

  Copy full SHA for 1cd4672

Commits on Oct 15, 2023

• Merge branch 'main' into cuda_nhwc

  gedoensmax

  authored
  33a87f8

  Copy full SHA for 33a87f8