Skip to content

Commit 476615b

Browse files
author
Bernhard Kerbl
committed
clean submodule
1 parent a25e54d commit 476615b

File tree

2 files changed

+2
-7
lines changed

2 files changed

+2
-7
lines changed

diff_gaussian_rasterization/rasterizer.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
from typing import NamedTuple
22
import torch.nn as nn
33
import torch
4-
import utils.profiling_utils as profiling_utils
5-
from diff_gaussian_rasterization import _C
4+
from . import _C
65

76
def rasterize_gaussians(
87
means3D,
@@ -97,13 +96,9 @@ def backward(ctx, grad_out_color, _):
9796
raster_settings.sh_degree,
9897
raster_settings.campos)
9998

100-
back_rng = profiling_utils.start("rasterize", "yellow", "render" )
101-
10299
# Compute gradients for relevant tensors by invoking backward method
103100
grad_means2D, grad_colors_precomp, grad_opacities, grad_means3D, grad_cov3Ds_precomp, grad_sh, grad_scales, grad_rotations = _C.rasterize_gaussians_backward(*args)
104101

105-
profiling_utils.stop(back_rng)
106-
107102
grads = (
108103
grad_means3D,
109104
grad_means2D,

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@
1010
CUDAExtension(
1111
name="diff_gaussian_rasterization._C",
1212
sources=[
13-
"rasterize_points.cu",
1413
"cuda_rasterizer/rasterizer_impl.cu",
1514
"cuda_rasterizer/forward.cu",
1615
"cuda_rasterizer/backward.cu",
16+
"rasterize_points.cu",
1717
"ext.cpp"],
1818
extra_compile_args={"nvcc": ["-I" + os.path.join(os.path.dirname(os.path.abspath(__file__)), "third_party/glm/")],
1919
"cxx": ["/wd4624"]})

0 commit comments

Comments
 (0)