format

Author: PPPoint-t

python/infinicore/nn/functional/glu.py

@@ -4,8 +4,7 @@
 
 
 def glu(input: Tensor, dim: int = -1) -> Tensor:
-    
     if infinicore.use_ntops and input.device.type in ("cuda", "musa"):
         return infinicore.ntops.torch.glu(input, dim)
 
-    return Tensor(_infinicore.glu(input._underlying, dim))
+    return Tensor(_infinicore.glu(input._underlying, dim))

python/infinicore/nn/functional/nll_loss.py

@@ -12,7 +12,6 @@ def nll_loss(
     *,
     out=None,
 ) -> Tensor:
-    
     if infinicore.use_ntops and input.device.type in ("cuda", "musa"):
         return infinicore.ntops.torch.nll_loss(
             input, target, weight=weight, ignore_index=ignore_index, reduction=reduction
@@ -23,18 +22,15 @@ def nll_loss(
     if out is None:
         return Tensor(
             _infinicore.nll_loss(
-                input._underlying, 
-                target._underlying, 
-                weight_underlying, 
-                ignore_index
+                input._underlying, target._underlying, weight_underlying, ignore_index
             )
         )
-    
+
     _infinicore.nll_loss_(
-        input._underlying, 
-        target._underlying, 
-        weight_underlying, 
-        out._underlying, 
-        ignore_index
+        input._underlying,
+        target._underlying,
+        weight_underlying,
+        out._underlying,
+        ignore_index,
     )
-    return out
+    return out

python/infinicore/ops/gcd.py

@@ -5,12 +5,12 @@
 
 def gcd(input: Tensor, other: Tensor, *, out=None) -> Tensor:
     r"""Computes the element-wise greatest common divisor (GCD)."""
-    
+
     if infinicore.use_ntops and input.device.type in ("cuda", "musa"):
         return infinicore.ntops.torch.gcd(input, other, out=out)
 
     if out is None:
         return Tensor(_infinicore.gcd(input._underlying, other._underlying))
-    
+
     _infinicore.gcd_(input._underlying, other._underlying, out._underlying)
-    return out
+    return out

python/infinicore/ops/gt.py

@@ -12,6 +12,6 @@ def gt(input: Tensor, other: Tensor | float, *, out: Tensor | None = None) -> Te
 
     if out is None:
         return Tensor(_infinicore.gt(input._underlying, other._underlying))
-    
+
     _infinicore.gt_(input._underlying, other._underlying, out._underlying)
-    return out
+    return out

python/infinicore/ops/select_scatter.py

@@ -7,4 +7,6 @@ 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)
 
-    return Tensor(_infinicore.select_scatter(input._underlying, src._underlying, dim, index))
+    return Tensor(
+        _infinicore.select_scatter(input._underlying, src._underlying, dim, index)
+    )