Code

[ROCM] Add Thrust support#7458

Code

Merged

masahi merged 10 commits intoapache:mainapache/tvm:mainfrom masahi:thrust-rocmmasahi/tvm:thrust-rocmCopy head branch name to clipboardFeb 17, 2021

+187-58

Commits

Commits on Feb 17, 2021

• enable rocm thrust, confrimed to work on sort and scan

  

  Masahiro Masuda

  committed
  4caad85

  Copy full SHA for 4caad85

• add rocm argsort strategy

  

  Masahiro Masuda

  committed
  b30d046

  Copy full SHA for b30d046

• Abort if CXX is not hipcc

  

  Masahiro Masuda

  committed
  2e0cefb

  Copy full SHA for 2e0cefb

• add more strategy

  

  Masahiro Masuda

  committed
  0a01d1b

  Copy full SHA for 0a01d1b

• add missing import

  

  Masahiro Masuda

  committed
  a6e6d39

  Copy full SHA for a6e6d39

• fix lint

  

  Masahiro Masuda

  committed
  2700105

  Copy full SHA for 2700105

• show supported data type in err msg

  

  Masahiro Masuda

  committed
  f9e60a8

  Copy full SHA for f9e60a8

• try remove rocthrust

  

  Masahiro Masuda

  committed
  684ecc3

  Copy full SHA for 684ecc3

• add missing include for rocthrust

  

  Masahiro Masuda

  committed
  3a11204

  Copy full SHA for 3a11204

• more minor change

  

  Masahiro Masuda

  committed
  2ee3d61

  Copy full SHA for 2ee3d61