Edward Z. Yang
|
acecd4acc1
|
Make torch.library decorators return function
This allows for chaining.
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78996
Approved by: https://github.com/Lezcano, https://github.com/Chillee
|
2022-06-08 01:57:00 +00:00 |
|
anjali411
|
79ddc32b6a
|
Add a check to ensure input func to Library.impl is callable
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77990
Approved by: https://github.com/albanD
|
2022-06-02 16:55:39 +00:00 |
|
anjali411
|
5984bc8233
|
Allow specifying alias analysis while registering new ops
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77690
Approved by: https://github.com/ezyang
|
2022-05-19 21:11:40 +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 |
|
anjali411
|
3d28ab0709
|
Minor follow up fixes for python registration
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76892
Approved by: https://github.com/albanD
|
2022-05-05 13:46:48 +00:00 |
|
anjali411
|
07f766df54
|
Allow creating new libraries and defining new operators from Python
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76250
Approved by: https://github.com/ezyang
|
2022-05-05 03:33:08 +00:00 |
|
anjali411
|
55f55a4cf6
|
Allow users to override kernels for existing C++ ops through Python
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75905
Approved by: https://github.com/ezyang
|
2022-05-05 03:31:39 +00:00 |
|