Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit d92492e

Browse files
authored
Correct Custom Module Name (#1180)
The torch extension name should be "lltm_cpp" instead of "lltm".
1 parent c15c405 commit d92492e

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

advanced_source/cpp_extension.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ For our extensions, the necessary binding code spans only four lines:
326326
327327
One bit to note here is the macro ``TORCH_EXTENSION_NAME``. The torch extension
328328
build will define it as the name you give your extension in the ``setup.py``
329-
script. In this case, the value of ``TORCH_EXTENSION_NAME`` would be "lltm".
329+
script. In this case, the value of ``TORCH_EXTENSION_NAME`` would be "lltm_cpp".
330330
This is to avoid having to maintain the name of the extension in two places
331331
(the build script and your C++ code), as a mismatch between the two can lead to
332332
nasty and hard to track issues.

0 commit comments

Comments
 (0)