extension-ffi
extension-ffi copied to clipboard
Examples of C extensions for PyTorch
I am following the example to create a PIP-distribuable package ```sh $ python setup.py sdist $ pip install dist/my_package-0.1.tar.gz ``` but I get this error ``` Processing ./dist/my_package-0.1.tar.gz Complete output...
This is pull request was created automatically because we noticed your project was missing a Code of Conduct file. Code of Conduct files facilitate respectful and constructive communities by establishing...
This is pull request was created automatically because we noticed your project was missing a Contributing file. CONTRIBUTING files explain how a developer can contribute to the project - which...
How can I be in c + + compiler environment, testing and validation. there the ‘THFloatTensor_data’ is not defined, when i use the code as follow: ``` int my_lib_add_forward(THFloatTensor *input1,...
the output‘nums of backward() shouldn’t be same as inputs‘nums of forward()??
We are doing a major refactor of the underlying TH libraries, which means that most of the C API will no longer work. Please see https://github.com/pytorch/extension-cpp for the replacement.
I build this demo according to README, but ran into the following error: c:\program files\nvidia gpu computing toolkit\cuda\v9.1\include\cuda_fp16.hpp(122): note: 参见“__half”的声明 C:\Program Files (x86)\Microsoft Visual Studio\2017\BuildTools\VC\Tools\MSVC\14.13.26128\bin\HostX86\x64\link.exe /nologo /INCREMENTAL:NO /LTCG /DLL /MANIFEST:EMBED,ID=2...
when i compile with c++, it print the error as follows: ```python /tmp/ccSlTwwt.o:在函数‘mylib(THTensor*)’中: /media/chenjun/data/C/daytwo/hello.cpp:8:对‘THFloatTensor_data’未定义的引用 collect2: error: ld returned 1 exit status The terminal process terminated with exit code: 1 ```
hello,where can i get the APIs of pytorch C extension function,such as THFloatTensor_resizeAs, THFloatTensor_add. Except those,has any function else? Thx.
Hello. Thanks for sharing the example. However, I am writing a extension that need many parameters, I think pass the parameters as c struct should be a method to make...