Code

TensorCore Support using Intrinsic#4136

Code

Merged

Laurawly merged 12 commits intoapache:masterapache/tvm:masterfrom Hzfengsy:masterHzfengsy/tvm:masterCopy head branch name to clipboardOct 24, 2019

+1,274-7

Commits

Commits on Oct 21, 2019

• add tensor core support

  Hzfengsy

  committed
  2842800

• avoid memory bank conflict

  Hzfengsy

  committed
  052d243

• fix thread sync & better performance

  Hzfengsy

  committed
  78133e3

• better performance

  Hzfengsy

  committed
  e290b1e

• add schedule test for conv2d

  Hzfengsy

  committed
  5caec7d

• extend into BatchMatMul

  Hzfengsy

  committed
  edea7df

• support config fragment shape and layout using intrinsic

  Hzfengsy

  committed
  cfff095

• add TensorCore tutorial

  Hzfengsy

  committed
  190d936

• add int support and fix lint

  Hzfengsy

  committed
  46f1b6e

Commits on Oct 22, 2019

• address comment

  Hzfengsy

  committed
  abe1e90

• add 32*16*8 TensorCore test

  Hzfengsy

  committed
  6104101

• fix wmma include logic

  

  Aleksander Fedorov

  authored and

  Hzfengsy

  committed
  f15daf0