xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-05-03 03:02:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update rasterizer_gpu.cu

Browse Source
This commit is contained in:
Andrej 2025-01-28 18:57:21 +01:00 committed by GitHub
parent 9c8cef2590
commit 23c096572d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
hy3dgen/texgen/custom_rasterizer/lib/custom_rasterizer_kernel/rasterizer_gpu.cu
View File

@ -18,7 +18,7 @@ __device__ void rasterizeTriangleGPU(int idx, float* vt0, float* vt1, float* vt2
if (isBarycentricCoordInBounds(baryCentricCoordinate)) { if (isBarycentricCoordInBounds(baryCentricCoordinate)) {
int pixel = py * width + px; int pixel = py * width + px;
if (zbuffer == 0) { if (zbuffer == 0) {
atomicExch(&zbuffer[pixel], (INT64)(idx + 1)); atomicExch((unsigned long long*)&zbuffer[pixel],(unsigned long long)(idx + 1));
continue; continue;
} }
float depth = baryCentricCoordinate[0] * vt0[2] + baryCentricCoordinate[1] * vt1[2] + baryCentricCoordinate[2] * vt2[2]; float depth = baryCentricCoordinate[0] * vt0[2] + baryCentricCoordinate[1] * vt1[2] + baryCentricCoordinate[2] * vt2[2];
@ -31,7 +31,7 @@ __device__ void rasterizeTriangleGPU(int idx, float* vt0, float* vt1, float* vt2
INT64 token = (INT64)z_quantize * MAXINT + (INT64)(idx + 1); INT64 token = (INT64)z_quantize * MAXINT + (INT64)(idx + 1);
if (depth < depth_thres) if (depth < depth_thres)
continue; continue;
atomicMin(&zbuffer[pixel], token); atomicMin((unsigned long long*)&zbuffer[pixel],(unsigned long long)token);
} }
} }
} }