Skip to content

Commit 01e7c5d

Browse files
committed
Fix SegmentedTensorProduct.flop_cost for 0 inputs (1 operand)
1 parent d1282a5 commit 01e7c5d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

cuequivariance/cuequivariance/segmented_tensor_product/segmented_tensor_product.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -514,8 +514,7 @@ def flop_cost(self, operand: int, algorithm: str = "optimal") -> int:
514514
d = self.move_operand_last(operand)
515515
subscripts = (
516516
d.coefficient_subscripts
517-
+ ","
518-
+ ",".join(operand.subscripts for operand in d.operands[:-1])
517+
+ "".join("," + operand.subscripts for operand in d.operands[:-1])
519518
+ "->"
520519
+ d.operands[-1].subscripts
521520
)

0 commit comments

Comments
 (0)