[2026春季][ T1-1-9] zili2004

src/ntops/kernels/__init__.py

@@ -14,6 +14,9 @@
     dropout,
     eq,
     exp,
+    frac,
+    fractional_max_pool2d,
+    fractional_max_pool3d,
     ge,
     gelu,
     gt,
@@ -25,13 +28,15 @@
     max_pool2d,
     mm,
     mul,
+    multilabel_margin_loss,
     ne,
     neg,
     pow,
     relu,
     rms_norm,
     rotary_position_embedding,
     rsqrt,
+    scatter_add,
     scaled_dot_product_attention,
     sigmoid,
     silu,
@@ -57,6 +62,9 @@
     "dropout",
     "eq",
     "exp",
+    "frac",
+    "fractional_max_pool2d",
+    "fractional_max_pool3d",
     "ge",
     "gelu",
     "gt",
@@ -68,13 +76,15 @@
     "max_pool2d",
     "mm",
     "mul",
+    "multilabel_margin_loss",
     "ne",
     "neg",
     "pow",
     "relu",
     "rms_norm",
     "rotary_position_embedding",
     "rsqrt",
+    "scatter_add",
     "scaled_dot_product_attention",
     "sigmoid",
     "silu",

src/ntops/kernels/frac.py

@@ -0,0 +1,24 @@
+import functools
+
+import ninetoothed.language as ntl
+from ninetoothed import Tensor
+
+from ntops.kernels.element_wise import arrangement
+
+
+def application(input, output):
+    fp32_input = (input * 2.0) / 2.0
+    floor_val = ntl.math.floor(fp32_input)
+    ceil_val = ntl.math.ceil(fp32_input)
+
+    # 获取截断后的整数部分
+    trunc_val = ntl.where(fp32_input >= 0, floor_val, ceil_val)
+    output = input - trunc_val  # noqa: F841
+
+
+def premake(ndim, dtype=None, block_size=None):
+    arrangement_ = functools.partial(arrangement, block_size=block_size)
+
+    tensors = (Tensor(ndim, dtype=dtype), Tensor(ndim, dtype=dtype))
+
+    return arrangement_, application, tensors