• Z
    Add sparse conv3d kernel (#39879) · bc99a76c
    zhangkaihuo 提交于
    * fix incorrect dims settings
    
    * sparse conv3d
    
    * fix out dims
    
    * test performance
    
    * test large shape success
    
    * opt scatter, double performance
    
    * test float16
    
    * remove profiling code
    
    * remove pten
    
    * opt code lines
    
    * correct boundary judgment
    
    * only cpu
    
    * test ci
    
    * test ci
    
    * remove the including paddle/fluid header; extract the conmmon function
    
    * opt code lines
    
    * use DenseTensor::data() instead of mutable_data
    
    * return rulebook for backward
    
    * specify layout
    
    * rename:conv -> sparse_conv3d
    bc99a76c
sparse_utils_kernel.cu 23.4 KB