From 5ab25073a4d0f53c30f0b0f27108ff9deb04c484 Mon Sep 17 00:00:00 2001
From: WangZhen <23097963+0x45f@users.noreply.github.com>
Date: Mon, 6 Jan 2025 14:33:26 +0800
Subject: [PATCH] Polish code gen filename (#399)

---
 src/flag_gems/utils/pointwise_dynamic.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/flag_gems/utils/pointwise_dynamic.py b/src/flag_gems/utils/pointwise_dynamic.py
index db776e28..2c151e58 100644
--- a/src/flag_gems/utils/pointwise_dynamic.py
+++ b/src/flag_gems/utils/pointwise_dynamic.py
@@ -1236,7 +1236,7 @@ def instantiate(self, ndim):
         # library, but we find generating a module simpler, since we can generating 2 functions
         # the kernel and the wrapper, and the wrapper calls the kernel.
         file_name = (
-            f"pointwise_dynamic_{self._scalar_fn_cache_key}_rank_{ndim}_"
+            f"pointwise_dynamic_{self._scalar_fn_cache_key}_{kernel_name}_"
             f"{'1d_tile_' if self.config.prefer_1d_tile else ''}"
             f"{'bptr_' if (not self.config.prefer_1d_tile and self.config.prefer_block_pointer) else ''}"
             f"pid_{self.pid}.py"