[KUNLUNXIN] Speed Up exp2 (#1254)

Author: Ason93

src/flag_gems/runtime/backend/_kunlunxin/ops/exp2.py

@@ -2,17 +2,27 @@
 
 import triton
 import triton.language as tl
+from _kunlunxin.utils.codegen_config_utils import CodeGenConfig
 
 from ..utils.pointwise_dynamic import pointwise_dynamic
 
-logger = logging.getLogger("flag_gems").getChild(__name__.lstrip("."))
+logger = logging.getLogger(__name__)
 
+config_ = CodeGenConfig(
+    512,
+    (65536, 65536, 65536),
+    32,
+    True,
+    prefer_1d_tile=True,
+    buffer_size_limit=4096,
+    unroll_num=8,
+)
 
-@pointwise_dynamic(promotion_methods=[(0, "INT_TO_FLOAT")])
+
+@pointwise_dynamic(promotion_methods=[(0, "INT_TO_FLOAT")], config=config_)
 @triton.jit
 def exp2_func(x):
-    LN2 = 0.69314718056
-    return tl.exp(x.to(tl.float32) * LN2)
+    return tl.exp2(x.to(tl.float32))
 
 
 def exp2(A):