Allow C64 and F32 buffers to be aliased (since a C64 tensor is just two F32 tensors stapled together).

James Spooner · Google-ML-Automation · commit 6c983a245ef8 · 2025-11-28T18:11:25.000-08:00
PiperOrigin-RevId: 837160497
diff --git a/xla/hlo/transforms/bfloat16_propagation.cc b/xla/hlo/transforms/bfloat16_propagation.cc
@@ -736,7 +736,7 @@ bool BFloat16Propagation::ResolveInconsistencyOfAliasingBuffersHelper(
           }
           VLOG(2) << "Adjust to F32 due to aliased dataflow value: "
                   << value->ToString() << "\n";
-          CHECK_EQ(value_type, F32);
+          CHECK(value_type == F32 || value_type == C64);
           type = F32;
           break;
         }

Original file line number	Diff line number	Diff line change
`@@ -736,7 +736,7 @@ bool BFloat16Propagation::ResolveInconsistencyOfAliasingBuffersHelper(`
`736`	`736`	`}`
`737`	`737`	`VLOG(2) << "Adjust to F32 due to aliased dataflow value: "`
`738`	`738`	`<< value->ToString() << "\n";`
`739`		`- CHECK_EQ(value_type, F32);`
	`739`	`+ CHECK(value_type == F32 \|\| value_type == C64);`
`740`	`740`	`type = F32;`
`741`	`741`	`break;`
`742`	`742`	`}`