Code

Kernel Storage V2#50

Code

Merged

xinyazhang merged 22 commits intomainROCm/aotriton:mainfrom xinyazhang/storagev2ROCm/aotriton:xinyazhang/storagev2Copy head branch name to clipboardNov 13, 2024

+802-351

Commits

Commits on Nov 7, 2024

• generate_compile: Add --test_clustering to figure out optimal clustering scheme

  xinyazhang

  committed
  eeaa826

• generate_compile: add option --generate_cluster_info

  Show description for c0d29cb
  xinyazhang

  committed
  c0d29cb

• build: support kernel storage v2

  Show description for 41abdc5
  xinyazhang

  committed
  41abdc5

• Add packed kernel support and defines AKS2 format (Untested).

  Show description for 2a30ad4
  xinyazhang

  committed
  2a30ad4

• Add the archiving tool aks2.py, also adjust the binary format

  xinyazhang

  committed
  d3ff51b

• Fix all compiling errors.

  xinyazhang

  committed
  0bcd77e

• Add some missing checks and cleans in packed_kernel.cc

  xinyazhang

  committed
  81ae205

• Install aotriton.images

  xinyazhang

  committed
  0c9a53e

• Fix in progress. python test/test_backward.py (Note NOT pytest) passed.

  xinyazhang

  committed
  1d0eceb

• Suppress debug output in Release build.

  xinyazhang

  committed
  50ef132

• Add cmake option AOTRITON_NOIMAGE_MODE

  Show description for d07e9b3
  xinyazhang

  committed
  d07e9b3

• Clang-format C++ Source/Header files

  xinyazhang

  committed
  2f17ff6

• clean up v2src/CMakeLists.txt

  xinyazhang

  committed
  f10ed75

• Clean up v2src/triton_kernel.cc

  xinyazhang

  committed
  1b0a0c2

• Copyright notices of aks2.py

  xinyazhang

  committed
  406c64b

• Fix the default of no image mode.

  xinyazhang

  committed
  bfbdea9

• Update Triton to Oct/23/2024

  xinyazhang

  committed
  bac1a06

• Make _common_test compatible with both PyTorch < 2.5 and >= 2.5

  xinyazhang

  committed
  e992093

• README: Update the Prerequisites

  Show description for 0dc5705
  xinyazhang

  committed
  0dc5705

Commits on Nov 8, 2024

• Remove AOTRITON_NO_SHARED option. It is incompatible with Kernel Storage V2.

  xinyazhang

  committed
  4ef62e4

Commits on Nov 12, 2024

• Add pybind11 to requirements.txt

  Show description for 536aee4
  xinyazhang

  committed
  536aee4

Commits on Nov 13, 2024

• add missing header <mutex> to packed_kernel.cc

  xinyazhang

  committed
  b7c33cd