Fix canonicalization error on sizes (#1494)

wsmoses · web-flow · commit bce12a6e1574 · 2025-10-21T09:33:35.000+09:00
diff --git a/src/enzyme_ad/jax/Dialect/Ops.cpp b/src/enzyme_ad/jax/Dialect/Ops.cpp
@@ -896,6 +896,9 @@ class LoadStorePointer2MemrefGEP final : public OpRewritePattern<T> {
       auto elemTy = gep.getElemType();
       if (elemTy.isIntOrFloat()) {
         gepElemSize = elemTy.getIntOrFloatBitWidth() / 8;
+      } else {
+        // Unknown type to get size from, bail early.
+        break;
       }
 
       gepOps.emplace_back(gep, gepElemSize);

Original file line number	Diff line number	Diff line change
`@@ -896,6 +896,9 @@ class LoadStorePointer2MemrefGEP final : public OpRewritePattern<T> {`
`896`	`896`	`auto elemTy = gep.getElemType();`
`897`	`897`	`if (elemTy.isIntOrFloat()) {`
`898`	`898`	`gepElemSize = elemTy.getIntOrFloatBitWidth() / 8;`
	`899`	`+ } else {`
	`900`	`+ // Unknown type to get size from, bail early.`
	`901`	`+ break;`
`899`	`902`	`}`
`900`	`903`
`901`	`904`	`gepOps.emplace_back(gep, gepElemSize);`