saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add TensorExpr Fuser tests (resubmit). (#35085)

Browse source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/35085

Test Plan: Imported from OSS

Differential Revision: D20552334

Pulled By: ZolotukhinM

fbshipit-source-id: 628fcf4719a879f18978ff8a0a64afbb045df645
This commit is contained in:
Mikhail Zolotukhin 2020-03-20 13:04:21 -07:00 committed by Facebook GitHub Bot
parent 3c409fc66c
commit 12f0052eee
3 changed files with 1011 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
test/run_test.py
View file

@ -70,6 +70,7 @@ TESTS = [
'test_jit_disabled',
'test_function_schema',
'test_overrides',
'test_jit_fuser_te',
]
# skip < 3.3 because mock is added in 3.3 and is used in rpc_spawn

1008
test/test_jit_fuser_te.py Normal file
View file

File diff suppressed because it is too large Load diff

2
torch/csrc/jit/python/init.cpp
View file

@ -344,6 +344,8 @@ void initJITBindings(PyObject* module) {
.def("_jit_pass_specialize_autogradzero", specializeAutogradZero)
.def("_jit_override_can_fuse_on_cpu", &overrideCanFuseOnCPU)
.def("_jit_override_can_fuse_on_gpu", &overrideCanFuseOnGPU)
.def("_jit_can_fuse_on_cpu", &canFuseOnCPU)
.def("_jit_can_fuse_on_gpu", &canFuseOnGPU)
.def("_jit_register_tensorexpr_fuser", &registerTensorExprFuser)
.def(
"_jit_differentiate",