pytorch/test/custom_operator/my_custom_ops2.py
Xuehai Pan 548c460bf1 [BE][Easy][7/19] enforce style for empty lines in import segments in test/[a-c]*/ and test/[q-z]*/ (#129758)
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
2024-07-31 10:54:03 +00:00

12 lines
212 B
Python

from model import get_custom_op_library_path
import torch
torch.ops.load_library(get_custom_op_library_path())
@torch.library.impl_abstract("custom::sin")
def sin_abstract(x):
return torch.empty_like(x)