KawasakiAkasei/Hunyuan3D_2.1_Low_VRAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

还原对于custom_rasterizer_kernel的修改

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
hy3dpaint/custom_rasterizer/lib/custom_rasterizer_kernel/grid_neighbor.cpp
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}; return {texture_positions, texture_feats, grid_neighbors, grid_downsamples, grid_evencorners, grid_oddcorners};
} }

2
hy3dpaint/custom_rasterizer/lib/custom_rasterizer_kernel/rasterizer.cpp
View File

@@ -136,4 +136,4 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
m.def("rasterize_image", &rasterize_image, "Custom image rasterization"); m.def("rasterize_image", &rasterize_image, "Custom image rasterization");
m.def("build_hierarchy", &build_hierarchy, "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"); m.def("build_hierarchy_with_feat", &build_hierarchy_with_feat, "Custom image rasterization");
} }

2
hy3dpaint/custom_rasterizer/lib/custom_rasterizer_kernel/rasterizer_gpu.cu
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>()); findices.data_ptr<int>(), (INT64*)z_min.data_ptr<long>(), width, height, num_vertices, num_faces, barycentric.data_ptr<float>());
return {findices, barycentric}; return {findices, barycentric};
} }