| from setuptools import setup | |
| from torch.utils.cpp_extension import BuildExtension, CUDAExtension | |
| setup(name='swin_window_process', | |
| ext_modules=[ | |
| CUDAExtension('swin_window_process', [ | |
| 'swin_window_process.cpp', | |
| 'swin_window_process_kernel.cu', | |
| ]) | |
| ], | |
| cmdclass={'build_ext': BuildExtension}) |
Xet Storage Details
- Size:
- 343 Bytes
- Xet hash:
- bcb404d92f28fc4a5e758db2bcc327364ac4dc1c0ea2f8aada9af8b7f577df68
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.