Commit Graph

11 Commits

Author SHA1 Message Date
davidriazati@fb.com
ee4be5322b Fix lint in test_tensorexpr_pybind (#57869)
Summary:
Fixes #{issue number}

Pull Request resolved: https://github.com/pytorch/pytorch/pull/57869

Pulled By: driazati

Reviewed By: gmagogsfm

Differential Revision: D28303542

fbshipit-source-id: a4c77c1e7ee92c39fd1c7d88422728e2f1c31680
2021-05-07 15:58:21 -07:00
Horace He
b38f153d91 [nnc] Added NNC lowerings for t/transpose/permute/expand + other cleaning (#57426)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57426

Test Plan: Imported from OSS

Reviewed By: ngimel

Differential Revision: D28293191

Pulled By: Chillee

fbshipit-source-id: b8fc44299acf2569c11e87e1991a2b724434b15d
2021-05-07 15:38:56 -07:00
Mikhail Zolotukhin
839d549f8f [JIT] Add a pass for removing a first (self) argument from a graph if it is unused. (#57169)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57169

The pass is planned to be used in AOT pipeline, where we expect input
graphs to be functional. As such, these graphs should not use 'self'
argument even if it is present, and thus it can be remove safely.

Test Plan: Imported from OSS

Reviewed By: bertmaher

Differential Revision: D28128328

Pulled By: ZolotukhinM

fbshipit-source-id: a7dfbf7776682826100c8eb0fef982a2e81c2554
2021-05-03 20:02:25 -07:00
Mikhail Zolotukhin
3ad3d8bd3f [JIT] Add a pass for annotating graph with input types derived from sample inputs. (#57076)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57076

This pass is intended to be used in conjunction with shape propagation
pass: first we use sample inputs to specify shape info for graph inputs
and then we run shape-prop to infer shapes of intermediate values in the
graph.

Differential Revision: D28048290

Test Plan: Imported from OSS

Reviewed By: astaff

Pulled By: ZolotukhinM

fbshipit-source-id: 778d772e873d59d77af9f669f45dc44b9ee5e443
2021-05-03 20:01:15 -07:00
Hui Guo
50cb75edce [tensorexpr] Add python bindings for TensorExprKernel (#54450)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54450

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D27243175

Pulled By: huiguoo

fbshipit-source-id: 820cf0d6cd1dd984d4153628e0f419d234668c82
2021-04-01 02:11:32 -07:00
Mikhail Zolotukhin
a176c73ed5 [TensorExpr] Reland: "PyBind: bind ExternalCalls." (#53063)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53063

The problem was that a derived class was marked with "py::nodelete",
while the base class wasn't. Now they both are marked correctly.

Test Plan: Imported from OSS

Reviewed By: bertmaher

Differential Revision: D26737877

Pulled By: ZolotukhinM

fbshipit-source-id: 17d9d430651c8f695fc7b6bf6784e7719e20a4d2
2021-03-01 20:44:10 -08:00
Bert Maher
7d060735ca Back out "[TensorExpr] PyBind: bind ExternalCalls."
Summary: Original commit changeset: e1ea3b3630d1

Test Plan: Broke tests, e.g. T85754010

Reviewed By: ZolotukhinM

Differential Revision: D26727166

fbshipit-source-id: 42d2090bc55ec2982a4a08c38278c80617d5398a
2021-03-01 10:51:06 -08:00
Mikhail Zolotukhin
e63ec556bf [TensorExpr] PyBind: bind ExternalCalls. (#52905)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52905

Differential Revision: D26707491

Test Plan: Imported from OSS

Reviewed By: Chillee

Pulled By: ZolotukhinM

fbshipit-source-id: e1ea3b3630d115e3d81842895c62e22c4cb06fb8
2021-02-26 20:28:13 -08:00
albanD
9920ae665b Make te a hidden package for now (#51690)
Summary:
As discussed with suo , having it in `torch._C.XX` means that it automatically gets added to `torch.XX` which is unfortunate. Making it `torch._C._XX` means that it won't be added to `torch.`.

Let me know if that approach to hide it is not good and we can update that.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51690

Reviewed By: gchanan

Differential Revision: D26243207

Pulled By: albanD

fbshipit-source-id: 3eb91a96635e90a6b98df799e3a732833dd280d5
2021-02-04 07:58:38 -08:00
Rong Rong (AI Infra)
806010b75e [BE] move more unittest.main() to run_tests() (#50923)
Summary:
Relate to https://github.com/pytorch/pytorch/issues/50483.

Everything except ONNX, detectron and release notes tests are moved to use common_utils.run_tests() to ensure CI reports XML correctly.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/50923

Reviewed By: samestep

Differential Revision: D26027621

Pulled By: walterddr

fbshipit-source-id: b04c03f10d1fe96181b720c4c3868e86e4c6281a
2021-01-25 07:23:09 -08:00
Mikhail Zolotukhin
e9dc8fc162 [TensorExpr] Add python bindings. (#49698)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49698

Reincarnation of #47620 by jamesr66a.

It's just an initial bunch of things that we're exposing to python, more
is expected to come in future. Some things can probably be done better,
but I'm putting this out anyway, since some other people were interested
in using and/or developing this.

Differential Revision: D25668694

Test Plan: Imported from OSS

Reviewed By: bertmaher

Pulled By: ZolotukhinM

fbshipit-source-id: fb0fd1b31e851ef9ab724686b9ac2d172fa4905a
2021-01-14 21:02:47 -08:00