Skip to content

Commit e3d7529

Browse files
authored
Merge pull request pytorch#12 from SeanNaren/master
Fix call in FFI tutorial
2 parents 9685a51 + 4cdaab5 commit e3d7529

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

Creating Extensions using FFI.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,10 @@ Now, you'll need a super short file, that will build your custom extension:
4747

4848
```python
4949
# build.py
50-
from torch.utils.ffi import compile_extension
51-
compile_extension(
50+
from torch.utils.ffi import create_extension
51+
create_extension(
5252
name='_ext.my_lib',
53-
header='src/my_lib.h',
53+
headers='src/my_lib.h',
5454
sources=['src/my_lib.c'],
5555
with_cuda=False
5656
)

0 commit comments

Comments
 (0)