Code

Collection Support [Inprogress]#802

Code

Closed

inocsin wants to merge 22 commits intopytorch:masterpytorch/TensorRT:masterfrom inocsin:collectioninocsin/Torch-TensorRT:collectionCopy head branch name to clipboard

+1,097-150

Commits

Commits on Apr 12, 2022

• feat: [collection] make torch_tensorrt::core::ir::Input and torch_tensorrt::Input compatible with IValue. Support simple case of tuple input model. Add unit test.

  Show description for 4434553
  inocsin

  committed
  4434553

• feat: [collection] try to defer determing the data type of tuple/list elements. Using two level vector to store ir::Input

  Show description for 2fc1363
  inocsin

  committed
  2fc1363

• feat: [collection] limited support for tuple input

  Show description for 0072e37
  inocsin

  committed
  0072e37

• fix: [collection] test normal input, fix bug

  Show description for b1d66cb
  inocsin

  committed
  b1d66cb

• feat: [collection] support list input type

  Show description for d4e54f1
  inocsin

  committed
  d4e54f1

• feat: [collection] support user defined input data type

  Show description for a9aa2e7
  inocsin

  committed
  a9aa2e7

• feat: [collection] support output type of list and tuple

  Show description for 5830cbe
  inocsin

  committed
  5830cbe

• feat: [collection] add unit test for complex collection model

  Show description for 6733cfb
  inocsin

  committed
  6733cfb

• chore: [collection] delete comments

  Show description for d21b0ab
  inocsin

  committed
  d21b0ab

• chore: [collection] update code and comments

  Show description for eada66d
  inocsin

  committed
  eada66d

• chore: [collection] rename ConversionInfo.collection_inputs to ConversionInfo.collection_input_spec_map

  Show description for 633c00f
  inocsin

  committed
  633c00f

• refactor: [collection] fuse Input with GraphInputs

  Show description for 89665c8
  inocsin

  committed
  89665c8

• feat: [collection] move collection test model to hub.py

  Show description for 205452e
  inocsin

  committed
  205452e

• test: [collection] update model path in test_collection.cpp

  Show description for a4d4131
  inocsin

  committed
  a4d4131

• fix: [collection] solve confict in ir.cpp

  Show description for 2d585e5
  inocsin

  committed
  2d585e5

• feat: [collection] update python api, refactor code

  Show description for 5f36810
  inocsin

  committed
  5f36810

• fix: [collection] remove aten::__getitem__ and prim::ListConstruct

  Show description for d9d8665
  inocsin

  committed
  d9d8665

• [collection] rebase to master, update some api

  Show description for 991f023
  inocsin

  committed
  991f023

Commits on Apr 14, 2022

• feat: [collection] handle prim::ListConstruct without fallback it manually

  Show description for 016c991
  inocsin

  committed
  016c991

• chore: [collection] update test_resolve_nontensor_inputs.cpp

  Show description for 2e7cd58
  inocsin

  committed
  2e7cd58

• fix: [collection] handle the case that only the output is collection and all the nodes can be converted

  Show description for b35cdd0
  inocsin

  committed
  b35cdd0

Commits on Apr 19, 2022

• fix: [collection] update tests/cpp/test_example_tensors.cpp

  Show description for fa6c10e
  inocsin

  committed
  fa6c10e