Skip to content

Commit c0aa1a9

Browse files
committed
Link to FAQ on main page in custom ops tutorial
1 parent 4779db8 commit c0aa1a9

File tree

1 file changed

+3
-32
lines changed

1 file changed

+3
-32
lines changed

advanced_source/torch_script_custom_ops.rst

+3-32
Original file line numberDiff line numberDiff line change
@@ -874,7 +874,9 @@ lines between Python, TorchScript and C++ to blend smoothly.
874874
875875
As always, if you run into any problems or have questions, you can use our
876876
`forum <https://discuss.pytorch.org/>`_ or `GitHub issues
877-
<https://github.com/pytorch/pytorch/issues>`_ to get in touch.
877+
<https://github.com/pytorch/pytorch/issues>`_ to get in touch. Also, our
878+
`frequently asked questions (FAQ) page
879+
<https://pytorch.org/cppdocs/notes/faq.html>`_ may have helpful information.
878880
879881
Appendix A: More Ways of Building Custom Operators
880882
--------------------------------------------------
@@ -1076,34 +1078,3 @@ visible to TorchScript:
10761078
>>> torch.ops.load_library("warp_perspective.so")
10771079
>>> print(torch.ops.custom.warp_perspective)
10781080
<built-in method custom::warp_perspective of PyCapsule object at 0x7ff51c5b7bd0>
1079-
1080-
1081-
FAQ
1082-
---------------------------------------
1083-
1084-
1085-
Custom Ops Types
1086-
*******************
1087-
1088-
TorchScript custom Ops supports a limited number of types. The types supported by JIT custom ops are declared `here<https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/core/jit_type.h>`_.
1089-
1090-
Currently, the types that are supported are:
1091-
1092-
- torch::Tensor
1093-
- at::Scalar
1094-
1095-
This represents the python number
1096-
1097-
- double
1098-
1099-
Note that float is not currently supported.
1100-
1101-
- int64_t
1102-
1103-
Note that int is not currently supported.
1104-
1105-
- bool
1106-
- string
1107-
- std::vector<torch::Tensor>
1108-
- std::vector<double>
1109-
- std::vector<int64_t>

0 commit comments

Comments
 (0)