Edward Z. Yang
83d40a4dba
linalg_cholesky_ex meta function
...
Taken from https://github.com/albanD/subclass_zoo/blob/main/python_meta_tensor.py
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78604
Approved by: https://github.com/bdhirsh , https://github.com/ngimel , https://github.com/Lezcano
2022-06-03 23:11:02 +00:00
Edward Z. Yang
6120a8e05d
Implement meta function for aten::index.Tensor
...
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78527
Approved by: https://github.com/bdhirsh , https://github.com/ngimel , https://github.com/Lezcano
2022-06-03 23:11:02 +00:00
Edward Z. Yang
1bd21dd152
_linalg_qr_helper meta function
...
Taken from https://github.com/albanD/subclass_zoo/blob/main/python_meta_tensor.py
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78603
Approved by: https://github.com/Lezcano , https://github.com/mruberry
2022-06-03 20:27:05 +00:00
Edward Z. Yang
9446f9678a
repeat_interleaves meta function
...
Taken from https://github.com/albanD/subclass_zoo/blob/main/python_meta_tensor.py
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78602
Approved by: https://github.com/mruberry
2022-06-02 21:24:46 +00:00
Edward Z. Yang
a1765f0176
addr ref
...
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78014
Approved by: https://github.com/ngimel
2022-05-25 01:40:11 +00:00
Edward Z. Yang
9e0e949484
Fix bugs, increase meta coverage
...
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77499
Approved by: https://github.com/mruberry
2022-05-19 21:04:57 +00:00
Edward Z. Yang
baeffdbc6c
reflection_pad2d support
...
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77681
Approved by: https://github.com/mruberry
2022-05-19 14:43:35 +00:00
Edward Z. Yang
d5ed73badd
Make it possible to register decompositions to Meta key
...
Decompositions can be used to fill in meta support where necessary,
assuming the operations they decompose to support meta key.
This PR adds register_meta kwarg to register_decomposition that
optionally lets you register the meta to the C++ dispatch table
for meta tensors. I use this to then get the meta function for
where and huber_loss for free.
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77353
Approved by: https://github.com/mruberry
2022-05-12 23:20:16 +00:00
anjali411
767af8e335
Add meta tensor support for some operations using python registration
...
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76916
Approved by: https://github.com/ezyang
2022-05-10 17:55:06 +00:00
