还原对于custom_rasterizer_kernel的修改

This commit is contained in:
qinmaohui
2025-09-10 22:37:29 +08:00
parent 928f41b289
commit 663ee27446
3 changed files with 3 additions and 3 deletions

View File

@@ -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};
}
}

View File

@@ -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");
}
}

View File

@@ -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};
}
}