frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

implement tuple constants (#31841)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/31841

Add Tuple Constants to JIT. The constraint here is that all elements of a tuple must themself be insertable as a a constant. Previously tuples were special cased in constant propagation, but now that there are more passes that are inserted constants, such as freezing, we should just have tuples be representable as constants.

Test Plan: Imported from OSS

Differential Revision: D19439514

Pulled By: eellison

fbshipit-source-id: 3810ba08ee349fa5598f4b53ea64525996637b1a
This commit is contained in:
Elias Ellison
2020-01-22 12:09:46 -08:00
committed by Facebook Github Bot
parent 69492ad6ac
commit 38d122eca9
11 changed files with 336 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/cpp/jit/test_irparser.cpp
View File

@ -151,8 +151,8 @@ graph(%x):
graph(%0 : Tensor,
%1 : Tensor,
%2 : Tensor):
%3 : int, %4 : Tensor = qqq::qqq[i_asdf=2, f_asdf=3.14, s_asdf="hello", ss_asdf=["hello world", "bye bye"]](%0)
%5 : int, %6 : Tensor = ppp::ppp[i_asdf=2, f_asdf=3.14, s_asdf="\"\"\"\"\nhe\"llo", q=[3, 2, 4]](%0)
%3 : int, %4 : Tensor = qqq::qqq[i_asdf=2, f_asdf=3., s_asdf="hello", ss_asdf=["hello world", "bye bye"]](%0)
%5 : int, %6 : Tensor = ppp::ppp[i_asdf=2, f_asdf=3., s_asdf="\"\"\"\"\nhe\"llo", q=[3, 2, 4]](%0)
%7 : float = vvv::vvv[s_asdf="hello"](%0)
%8 : string = z::z()
return (%7)