commit b2a9c3baa979dcd432f23eadb188fc8787519deb
author Hao Lu <haolu7@gmail.com> Wed Sep 02 19:05:39 2020 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Wed Sep 02 19:07:35 2020 -0700
tree 60d460190c92414c84ce6c14b42dc7f3bf4d54a3
parent b2aaf212aa1a34ef5b9ce170510bd4d6f9f1181f

[TVM] Support fp16 weights in c2_frontend (#44070)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44070

Reviewed By: yinghai

Differential Revision: D23444253

fbshipit-source-id: 0bfa98172dfae835eba5ca7cbe30383ba964c2a6

caffe2/opt/tvm_transformer.cc

diff --git a/caffe2/opt/tvm_transformer.cc b/caffe2/opt/tvm_transformer.cc
index 2c62f65..39fd89d 100644
--- a/caffe2/opt/tvm_transformer.cc
+++ b/caffe2/opt/tvm_transformer.cc
@@ -194,6 +194,7 @@
       "DotProduct",
       "EnsureCPUOutput",
       "ExpandDims",
+      "FbFCPacked",
       "FC",
       "FCTransposed",
       "Flatten",