还原对于custom_rasterizer_kernel的修改
This commit is contained in:
@@ -571,4 +571,4 @@ std::vector<std::vector<torch::Tensor>> build_hierarchy_with_feat(
|
||||
}
|
||||
}
|
||||
return {texture_positions, texture_feats, grid_neighbors, grid_downsamples, grid_evencorners, grid_oddcorners};
|
||||
}
|
||||
}
|
||||
|
||||
@@ -136,4 +136,4 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
|
||||
m.def("rasterize_image", &rasterize_image, "Custom image rasterization");
|
||||
m.def("build_hierarchy", &build_hierarchy, "Custom image rasterization");
|
||||
m.def("build_hierarchy_with_feat", &build_hierarchy_with_feat, "Custom image rasterization");
|
||||
}
|
||||
}
|
||||
|
||||
@@ -124,4 +124,4 @@ std::vector<torch::Tensor> rasterize_image_gpu(torch::Tensor V, torch::Tensor F,
|
||||
findices.data_ptr<int>(), (INT64*)z_min.data_ptr<long>(), width, height, num_vertices, num_faces, barycentric.data_ptr<float>());
|
||||
|
||||
return {findices, barycentric};
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user