Code

[WIP] GNN Cost Model addition#17551

Code

Draft

dwijenchawra wants to merge 9 commits intoapache:mainapache/tvm:mainfrom dwijenchawra:maindwijenchawra/tvm:mainCopy head branch name to clipboard

+1,099-1

Commits

Commits on Dec 3, 2024

• added code to xgboost so we can lower states on the fly

  dwijenchawra

  committed
  b43fc37

  Copy full SHA for b43fc37

Commits on Dec 4, 2024

• adding gnnmodel to auto_scheduler namespace

  dwijenchawra

  committed
  66b505c

  Copy full SHA for 66b505c

• 1 more namespace addition

  dwijenchawra

  committed
  a7ecd8f

  Copy full SHA for a7ecd8f

• extracting ast into graph working

  dwijenchawra

  committed
  4600a21

  Copy full SHA for 4600a21

• tvm objects not serializable :(

  dwijenchawra

  committed
  1ab4822

  Copy full SHA for 1ab4822

• parallel lowering and ast generation in python

  dwijenchawra

  committed
  6370f18

  Copy full SHA for 6370f18

• Merge branch 'apache:main' into main

  dwijenchawra

  authored
  cb6405d

  Copy full SHA for cb6405d

• remove unused files

  dwijenchawra

  committed
  8214ebf

  Copy full SHA for 8214ebf

Commits on Dec 5, 2024

• updates

  dwijenchawra

  committed
  2c3582d

  Copy full SHA for 2c3582d