cpp_extension dist must be a Distribution instance

dist must be a Distribution instance

原因是参数传的不对 参数应该是Distribution 实例

有两种写法，但是都不对

ffi = cpp_extension.build_ext(
 

  

   

    

     

    
    

     

      ffi = cpp_extension.BuildExtension(
     
    
   

    

     

    
    

     

       '_ext.dcn_v2',
     
    
   

    

     

    
    

     

       headers=headers,
     
    
   

    

     

    
    

     

       sources=sources,
     
    
   

    

     

    
    

     

       define_macros=defines,
     
    
   

    

     

    
    

     

       relative_to=__file__,
     
    
   

    

     

    
    

     

       with_cuda=with_cuda,
     
    
   

    

     

    
    

     

       extra_objects=extra_objects,
     
    
   

    

     

    
    

     

       # extra_compile_args=extra_compile_args
     
    
   

    

     

    
    

     

      )
     
    
  
 

c++用法：

  

   

    

     

    
    

     

      from setuptools import setup
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from torch.utils.cpp_extension import BuildExtension, CUDAExtension, CppExtension
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if torch.cuda.is_available():
     
    
   

    

     

    
    

     

       print('Including CUDA code.')
     
    
   

    

     

    
    

     

       setup(
     
    
   

    

     

    
    

     

       name='trilinear',
     
    
   

    

     

    
    

     

       ext_modules=[
     
    
   

    

     

    
    

     

       CUDAExtension('trilinear', [
     
    
   

    

     

    
    

     
 'src/trilinear_cuda.cpp',
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/90577512