├── .gitignore ├── README.md ├── package ├── README.md ├── build.py ├── my_package │ ├── __init__.py │ ├── functions │ │ ├── __init__.py │ │ └── add.py │ ├── modules │ │ ├── __init__.py │ │ └── add.py │ └── src │ │ ├── my_lib.c │ │ ├── my_lib.h │ │ ├── my_lib_cuda.c │ │ └── my_lib_cuda.h ├── setup.py └── test │ └── test.py └── script ├── README.md ├── build.py ├── functions ├── __init__.py └── add.py ├── modules ├── __init__.py └── add.py ├── src ├── my_lib.c ├── my_lib.h ├── my_lib_cuda.c └── my_lib_cuda.h └── test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/README.md -------------------------------------------------------------------------------- /package/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/README.md -------------------------------------------------------------------------------- /package/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/build.py -------------------------------------------------------------------------------- /package/my_package/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/my_package/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/my_package/functions/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/my_package/functions/add.py -------------------------------------------------------------------------------- /package/my_package/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/my_package/modules/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/my_package/modules/add.py -------------------------------------------------------------------------------- /package/my_package/src/my_lib.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/my_package/src/my_lib.c -------------------------------------------------------------------------------- /package/my_package/src/my_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/my_package/src/my_lib.h -------------------------------------------------------------------------------- /package/my_package/src/my_lib_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/my_package/src/my_lib_cuda.c -------------------------------------------------------------------------------- /package/my_package/src/my_lib_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/my_package/src/my_lib_cuda.h -------------------------------------------------------------------------------- /package/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/setup.py -------------------------------------------------------------------------------- /package/test/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/package/test/test.py -------------------------------------------------------------------------------- /script/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/README.md -------------------------------------------------------------------------------- /script/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/build.py -------------------------------------------------------------------------------- /script/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /script/functions/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/functions/add.py -------------------------------------------------------------------------------- /script/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /script/modules/add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/modules/add.py -------------------------------------------------------------------------------- /script/src/my_lib.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/src/my_lib.c -------------------------------------------------------------------------------- /script/src/my_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/src/my_lib.h -------------------------------------------------------------------------------- /script/src/my_lib_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/src/my_lib_cuda.c -------------------------------------------------------------------------------- /script/src/my_lib_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/src/my_lib_cuda.h -------------------------------------------------------------------------------- /script/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pytorch/extension-ffi/HEAD/script/test.py --------------------------------------------------------------------------------