You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 lines
223 B
6 lines
223 B
from paddle.utils.cpp_extension import CppExtension, CUDAExtension, setup
|
|
|
|
if __name__ == "__main__":
|
|
setup(
|
|
name='rbox_iou_ops',
|
|
ext_modules=CUDAExtension(sources=['rbox_iou_op.cc', 'rbox_iou_op.cu']))
|
|
|