Code

QAT : TRT 8 compatible workflow#804

Code

Draft

SrivastavaKshitij wants to merge 33 commits intoNVIDIA-AI-IOT:masterNVIDIA-AI-IOT/torch2trt:masterfrom SrivastavaKshitij:kshitij_qat_trt8SrivastavaKshitij/torch2trt:kshitij_qat_trt8Copy head branch name to clipboard

+1,072-958

Commits

Commits on Aug 24, 2022

• made qat layer scriptable

  SrivastavaKshitij

  committed
  e66f210

  Copy full SHA for e66f210

Commits on Aug 25, 2022

• WIP - refactor qat library

  SrivastavaKshitij

  committed
  38342b2

  Copy full SHA for 38342b2

• fixed mapping function

  SrivastavaKshitij

  committed
  9922d1c

  Copy full SHA for 9922d1c

Commits on Sep 2, 2022

• WIP

  SrivastavaKshitij

  committed
  44d0750

  Copy full SHA for 44d0750

Commits on Sep 7, 2022

• WIP

  SrivastavaKshitij

  committed
  73e523b

  Copy full SHA for 73e523b

Commits on Sep 8, 2022

• added converter for quantconv

  SrivastavaKshitij

  committed
  5b1d6b9

  Copy full SHA for 5b1d6b9

Commits on Sep 9, 2022

• working quant conv converter

  SrivastavaKshitij

  committed
  f4b52ee

  Copy full SHA for f4b52ee

• changing workflow

  SrivastavaKshitij

  committed
  39244d7

  Copy full SHA for 39244d7

Commits on Sep 10, 2022

• removed redundant data

  SrivastavaKshitij

  committed
  39b8f9f

  Copy full SHA for 39b8f9f

Commits on Sep 11, 2022

• working maxpool layer

  SrivastavaKshitij

  committed
  fa6712c

  Copy full SHA for fa6712c

• added working quantmaxpool2d laayer

  SrivastavaKshitij

  committed
  3d260ac

  Copy full SHA for 3d260ac

Commits on Sep 12, 2022

• added support for AdaptiveAvgPool2d

  SrivastavaKshitij

  committed
  7419d70

  Copy full SHA for 7419d70

• fixed trt and non trt mode

  SrivastavaKshitij

  committed
  c9c3f0f

  Copy full SHA for c9c3f0f

• added converter for quant adaptive avgpool 2d

  SrivastavaKshitij

  committed
  4687282

  Copy full SHA for 4687282

• fixed adaptiveavgpool2d converter import

  SrivastavaKshitij

  committed
  c4e13ef

  Copy full SHA for c4e13ef

• fixed pytorch fake quant ops

  SrivastavaKshitij

  committed
  37f89fb

  Copy full SHA for 37f89fb

• added generic converter

  SrivastavaKshitij

  committed
  6e63c4f

  Copy full SHA for 6e63c4f

Commits on Sep 13, 2022

• fixed nn.conv2d

  SrivastavaKshitij

  committed
  179a9d9

  Copy full SHA for 179a9d9

• removed graphviz import

  SrivastavaKshitij

  committed
  f31f67c

  Copy full SHA for f31f67c

Commits on Sep 14, 2022

• fixed documentation

  SrivastavaKshitij

  committed
  60cb7e6

  Copy full SHA for 60cb7e6

Commits on Sep 15, 2022

• fixed build script

  SrivastavaKshitij

  committed
  c0521f1

  Copy full SHA for c0521f1

• fixed parameter type for torch.fake_quantize per channel

  SrivastavaKshitij

  committed
  3f91fae

  Copy full SHA for 3f91fae

• Fixed accuracy metrics

  SrivastavaKshitij

  committed
  1068c64

  Copy full SHA for 1068c64

• added new patch file

  SrivastavaKshitij

  committed
  fe5a74d

  Copy full SHA for fe5a74d

Commits on Sep 19, 2022

• fixed import warning

  SrivastavaKshitij

  committed
  5e3e65b

  Copy full SHA for 5e3e65b

Commits on Sep 22, 2022

• fixed precision of zero point

  SrivastavaKshitij

  committed
  0ef2df0

  Copy full SHA for 0ef2df0

Commits on Sep 23, 2022

• fixed logging issue for conv2d while scripting

  SrivastavaKshitij

  committed
  85cbc66

  Copy full SHA for 85cbc66

• stripped off tensor quantizer depedency, hopefully

  SrivastavaKshitij

  committed
  5f3de32

  Copy full SHA for 5f3de32

• added min reproducible file

  SrivastavaKshitij

  committed
  e0742c1

  Copy full SHA for e0742c1

Commits on Sep 24, 2022

• fixed converters

  SrivastavaKshitij

  committed
  3a3c4bd

  Copy full SHA for 3a3c4bd

Commits on Sep 26, 2022

• fixed loading

  SrivastavaKshitij

  committed
  e0c7e62

  Copy full SHA for e0c7e62

Commits on Sep 27, 2022

• fixed quant axis initial value

  SrivastavaKshitij

  committed
  f994bad

  Copy full SHA for f994bad

Commits on Oct 6, 2022

• WIP

  SrivastavaKshitij

  committed
  8317ffa

  Copy full SHA for 8317ffa