mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
548c460bf1
See https://github.com/pytorch/pytorch/pull/129751#issue-2380881501. Most changes are auto-generated by linter. You can review these PRs via: ```bash git diff --ignore-all-space --ignore-blank-lines HEAD~1 ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/129758 Approved by: https://github.com/ezyang
20 lines
462 B
Python
20 lines
462 B
Python
from model import get_custom_op_library_path
|
|
|
|
import torch
|
|
|
|
|
|
torch.ops.load_library(get_custom_op_library_path())
|
|
|
|
|
|
# NB: The impl_abstract_pystub for cos actually
|
|
# specifies it should live in the my_custom_ops2 module.
|
|
@torch.library.impl_abstract("custom::cos")
|
|
def cos_abstract(x):
|
|
return torch.empty_like(x)
|
|
|
|
|
|
# NB: There is no impl_abstract_pystub for tan
|
|
@torch.library.impl_abstract("custom::tan")
|
|
def tan_abstract(x):
|
|
return torch.empty_like(x)
|