Proper module

rasterizer.py → diff_gaussian_rasterization/rasterizer.py

@@ -2,7 +2,7 @@ from typing import NamedTuple
 import torch.nn as nn
 import torch
 import utils.profiling_utils as profiling_utils
-from diff_rasterizationCUDA import _C
+from diff_gaussian_rasterization import _C
 
 def rasterize_gaussians(
     means3D,

setup.py

@@ -4,10 +4,11 @@ import os
 os.path.dirname(os.path.abspath(__file__))
 
 setup(
-    name="diff_rasterizationCUDA",
+    name="diff_gaussian_rasterization",
+    packages=['diff_gaussian_rasterization'],
     ext_modules=[
         CUDAExtension(
-            name="diff_rasterizationCUDA._C",
+            name="diff_gaussian_rasterization._C",
             sources=[
             "rasterize_points.cu", 
             "cuda_rasterizer/rasterizer_impl.cu",