From 97eaf7f83f125e34b85a5ce49e91f7ea59e4f06d Mon Sep 17 00:00:00 2001
From: Zhuoyang <zhuoyangpan@outlook.com>
Date: Fri, 6 Oct 2023 02:37:00 -0700
Subject: [PATCH] fix the naming issue

---
 diff_rast/_torch_impl.py                                        | 2 +-
 ...sterize_forward_impl.py => test_rasterize_forward_kernel.py} | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)
 rename tests/{test_rasterize_forward_impl.py => test_rasterize_forward_kernel.py} (99%)

diff --git a/diff_rast/_torch_impl.py b/diff_rast/_torch_impl.py
index 55d083be7..b1b7cd35e 100644
--- a/diff_rast/_torch_impl.py
+++ b/diff_rast/_torch_impl.py
@@ -368,7 +368,7 @@ def bin_and_sort_gaussians(
     return isect_ids, gaussian_ids, isect_ids_sorted, gaussian_ids_sorted, tile_bins
 
 
-def rasterize_forward_impl(
+def rasterize_forward_kernel(
     tile_bounds,
     block,
     img_size,
diff --git a/tests/test_rasterize_forward_impl.py b/tests/test_rasterize_forward_kernel.py
similarity index 99%
rename from tests/test_rasterize_forward_impl.py
rename to tests/test_rasterize_forward_kernel.py
index 1f437cefa..3a4fc1cee 100644
--- a/tests/test_rasterize_forward_impl.py
+++ b/tests/test_rasterize_forward_kernel.py
@@ -89,7 +89,7 @@ def test_rasterize_forward_kernel():
         background,
     )
 
-    (_out_img, _final_Ts, _final_idx,) = _torch_impl.rasterize_forward_impl(
+    (_out_img, _final_Ts, _final_idx,) = _torch_impl.rasterize_forward_kernel(
         tile_bounds,
         block,
         img_size,