Code

Fix VNNI affine maps#998

Code

Merged

adam-smnk merged 26 commits intoplaidml:mainplaidml/tpp-mlir:mainfrom adam-smnk:fix-vnni-affine-mapsadam-smnk/tpp-mlir:fix-vnni-affine-mapsCopy head branch name to clipboardJan 10, 2025

+531-265

Commits

Commits on Jan 6, 2025

• WIP pack vnni rework

  adam-smnk

  committed
  62a0f90

  Copy full SHA for 62a0f90

• Complete vnni logic + refactor

  adam-smnk

  committed
  cc58473

  Copy full SHA for cc58473

• Use verifier without error msg

  adam-smnk

  committed
  83ac1c5

  Copy full SHA for 83ac1c5

• Update pass unit tests

  adam-smnk

  committed
  406dbba

  Copy full SHA for 406dbba

• Fix matcher - use dim positions instead of iterators

  adam-smnk

  committed
  32b56f4

  Copy full SHA for 32b56f4

• Update mlir-gen to new vnni maps

  adam-smnk

  committed
  2e243e1

  Copy full SHA for 2e243e1

• WIP test update

  adam-smnk

  committed
  938d12e

  Copy full SHA for 938d12e

• WIP linalg to xsmm update

  adam-smnk

  committed
  8b7f6e3

  Copy full SHA for 8b7f6e3

• VNNI util refactor + fix

  adam-smnk

  committed
  2aaf8ec

  Copy full SHA for 2aaf8ec

• Update brgemm vnni matcher

  adam-smnk

  committed
  6322548

  Copy full SHA for 6322548

• Update brgemm linalg to xsmm lowering

  adam-smnk

  committed
  1fc4d66

  Copy full SHA for 1fc4d66

• Don't transpose A in VNNI

  adam-smnk

  committed
  2886005

  Copy full SHA for 2886005

• Improve structure check

  adam-smnk

  committed
  a554bd5

  Copy full SHA for a554bd5

• Fix K dim size in dispatch

  adam-smnk

  committed
  632d029

  Copy full SHA for 632d029

• Fix batch dimension detection

  adam-smnk

  committed
  e4d7280

  Copy full SHA for e4d7280

• Update tests

  adam-smnk

  committed
  8d7a53a

  Copy full SHA for 8d7a53a

• Update tests

  adam-smnk

  committed
  404f574

  Copy full SHA for 404f574

• Remove remaining invalid floordiv 2 usage from tests

  adam-smnk

  committed
  d44f475

  Copy full SHA for d44f475

• Compare VNNI results from xsmm vs loops

  adam-smnk

  committed
  9f400b5

  Copy full SHA for 9f400b5

• Handle mismatch gracefully

  adam-smnk

  committed
  bade09e

  Copy full SHA for bade09e

Commits on Jan 7, 2025

• Improve docs + naming

  adam-smnk

  committed
  c8e421a

  Copy full SHA for c8e421a

Commits on Jan 9, 2025

• Refactor + docs

  adam-smnk

  committed
  ccafcf8

  Copy full SHA for ccafcf8

• Fix typo

  adam-smnk

  committed
  aac340a

  Copy full SHA for aac340a

• Add proper vnni check

  adam-smnk

  committed
  6513c14

  Copy full SHA for 6513c14

Commits on Jan 10, 2025

• mlir-gen shape asserts

  adam-smnk

  committed
  463b80b

  Copy full SHA for 463b80b

• Constrain vnni matcher maps

  adam-smnk

  committed
  709eaf8

  Copy full SHA for 709eaf8