File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -733,7 +733,7 @@ bool MarginCrossEntropyOpInferSymbolicShape(
733733 size_t logits_rank = logits_dims.size ();
734734 auto axis = logits_rank - 1 ;
735735
736- for (int i = 0 ; i < logits_rank; i++) {
736+ for (size_t i = 0 ; i < logits_rank; i++) {
737737 if (i != axis && op->attribute <pir::BoolAttribute>(" is_runtime" ).data ()) {
738738 infer_context->AddBroadcastableCstr (logits_dims[i], labels_dims[i]);
739739 }
@@ -759,7 +759,7 @@ bool MarginCrossEntropyOpInferSymbolicShape(
759759
760760 return true ;
761761}
762-
762+
763763// bool PullBoxSparseOpInferSymbolicShape(pir::Operation *op,
764764// pir::InferSymbolicShapeContext
765765// *infer_context) {
@@ -780,7 +780,7 @@ bool MarginCrossEntropyOpInferSymbolicShape(
780780// // pass
781781// return true;
782782// }
783-
783+
784784bool SearchsortedOpInferSymbolicShape (
785785 pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
786786 // The shape of output is the same as input `values` (op->operand_source(1))
You can’t perform that action at this time.
0 commit comments