Commit 476615b0 by Bernhard Kerbl

clean submodule

parent a25e54d7
from typing import NamedTuple from typing import NamedTuple
import torch.nn as nn import torch.nn as nn
import torch import torch
import utils.profiling_utils as profiling_utils from . import _C
from diff_gaussian_rasterization import _C
def rasterize_gaussians( def rasterize_gaussians(
means3D, means3D,
...@@ -97,13 +96,9 @@ class _RasterizeGaussians(torch.autograd.Function): ...@@ -97,13 +96,9 @@ class _RasterizeGaussians(torch.autograd.Function):
raster_settings.sh_degree, raster_settings.sh_degree,
raster_settings.campos) raster_settings.campos)
back_rng = profiling_utils.start("rasterize", "yellow", "render" )
# Compute gradients for relevant tensors by invoking backward method # Compute gradients for relevant tensors by invoking backward method
grad_means2D, grad_colors_precomp, grad_opacities, grad_means3D, grad_cov3Ds_precomp, grad_sh, grad_scales, grad_rotations = _C.rasterize_gaussians_backward(*args) grad_means2D, grad_colors_precomp, grad_opacities, grad_means3D, grad_cov3Ds_precomp, grad_sh, grad_scales, grad_rotations = _C.rasterize_gaussians_backward(*args)
profiling_utils.stop(back_rng)
grads = ( grads = (
grad_means3D, grad_means3D,
grad_means2D, grad_means2D,
......
...@@ -10,10 +10,10 @@ setup( ...@@ -10,10 +10,10 @@ setup(
CUDAExtension( CUDAExtension(
name="diff_gaussian_rasterization._C", name="diff_gaussian_rasterization._C",
sources=[ sources=[
"rasterize_points.cu",
"cuda_rasterizer/rasterizer_impl.cu", "cuda_rasterizer/rasterizer_impl.cu",
"cuda_rasterizer/forward.cu", "cuda_rasterizer/forward.cu",
"cuda_rasterizer/backward.cu", "cuda_rasterizer/backward.cu",
"rasterize_points.cu",
"ext.cpp"], "ext.cpp"],
extra_compile_args={"nvcc": ["-I" + os.path.join(os.path.dirname(os.path.abspath(__file__)), "third_party/glm/")], extra_compile_args={"nvcc": ["-I" + os.path.join(os.path.dirname(os.path.abspath(__file__)), "third_party/glm/")],
"cxx": ["/wd4624"]}) "cxx": ["/wd4624"]})
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment