Commit 476615b0 by Bernhard Kerbl

clean submodule

parent a25e54d7
from typing import NamedTuple
import torch.nn as nn
import torch
import utils.profiling_utils as profiling_utils
from diff_gaussian_rasterization import _C
from . import _C
def rasterize_gaussians(
means3D,
......@@ -97,13 +96,9 @@ class _RasterizeGaussians(torch.autograd.Function):
raster_settings.sh_degree,
raster_settings.campos)
back_rng = profiling_utils.start("rasterize", "yellow", "render" )
# 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)
profiling_utils.stop(back_rng)
grads = (
grad_means3D,
grad_means2D,
......
......@@ -10,10 +10,10 @@ setup(
CUDAExtension(
name="diff_gaussian_rasterization._C",
sources=[
"rasterize_points.cu",
"cuda_rasterizer/rasterizer_impl.cu",
"cuda_rasterizer/forward.cu",
"cuda_rasterizer/backward.cu",
"rasterize_points.cu",
"ext.cpp"],
extra_compile_args={"nvcc": ["-I" + os.path.join(os.path.dirname(os.path.abspath(__file__)), "third_party/glm/")],
"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