fromat

Author: PPPoint-t

python/infinicore/__init__.py

@@ -45,18 +45,18 @@
 from infinicore.ops.add import add
 from infinicore.ops.add_rms_norm import add_rms_norm, add_rms_norm_
 from infinicore.ops.attention import attention
+from infinicore.ops.gcd import gcd
+from infinicore.ops.gt import gt
 from infinicore.ops.matmul import matmul
 from infinicore.ops.mul import mul
 from infinicore.ops.narrow import narrow
 from infinicore.ops.paged_attention import paged_attention
 from infinicore.ops.paged_attention_prefill import paged_attention_prefill
 from infinicore.ops.paged_caching import paged_caching
 from infinicore.ops.rearrange import rearrange
+from infinicore.ops.select_scatter import select_scatter
 from infinicore.ops.squeeze import squeeze
 from infinicore.ops.unsqueeze import unsqueeze
-from infinicore.ops.gcd import gcd
-from infinicore.ops.gt import gt
-from infinicore.ops.select_scatter import select_scatter
 from infinicore.tensor import (
     Tensor,
     empty,

python/infinicore/nn/functional/__init__.py

@@ -1,13 +1,13 @@
 from .causal_softmax import causal_softmax
 from .embedding import embedding
+from .glu import glu
 from .linear import linear
+from .nll_loss import nll_loss
 from .random_sample import random_sample
 from .rms_norm import rms_norm
 from .rope import RopeAlgo, rope
 from .silu import silu
 from .swiglu import swiglu
-from .nll_loss import nll_loss
-from .glu import glu
 
 __all__ = [
     "causal_softmax",

python/infinicore/nn/functional/glu.py

@@ -2,6 +2,7 @@
 from infinicore.lib import _infinicore
 from infinicore.tensor import Tensor
 
+
 def glu(input: Tensor, dim: int = -1) -> Tensor:
 
     if infinicore.use_ntops and input.device.type in ("cuda", "musa"):

python/infinicore/nn/functional/nll_loss.py

@@ -2,6 +2,7 @@
 from infinicore.lib import _infinicore
 from infinicore.tensor import Tensor
 
+
 def nll_loss(
     input: Tensor,
     target: Tensor,

python/infinicore/ops/gcd.py

@@ -2,6 +2,7 @@
 from infinicore.lib import _infinicore
 from infinicore.tensor import Tensor
 
+
 def gcd(input: Tensor, other: Tensor, *, out=None) -> Tensor:
     r"""Computes the element-wise greatest common divisor (GCD)."""
 

python/infinicore/ops/gt.py

@@ -2,6 +2,7 @@
 from infinicore.lib import _infinicore
 from infinicore.tensor import Tensor
 
+
 def gt(input: Tensor, other: Tensor | float, *, out: Tensor | None = None) -> Tensor:
     if infinicore.use_ntops and input.device.type in ("cuda", "musa"):
         return infinicore.ntops.torch.gt(input, other, out=out)

python/infinicore/ops/select_scatter.py

@@ -2,6 +2,7 @@
 from infinicore.lib import _infinicore
 from infinicore.tensor import Tensor
 
+
 def select_scatter(input: Tensor, src: Tensor, dim: int, index: int) -> Tensor:
     if infinicore.use_ntops and input.device.type in ("cuda", "musa"):
         return infinicore.ntops.torch.select_scatter(input, src, dim, index)