diff --git a/iree/test/e2e/xla_ops/BUILD b/iree/test/e2e/xla_ops/BUILD
index b492e10..744ff22 100644
--- a/iree/test/e2e/xla_ops/BUILD
+++ b/iree/test/e2e/xla_ops/BUILD
@@ -62,6 +62,7 @@
         # "gather.mlir",
         # "gather_concat.mlir",
         #
+        "iota.mlir",
         "log.mlir",
         "maximum.mlir",
         "minimum.mlir",
@@ -106,6 +107,7 @@
         "divide.mlir",
         "dot.mlir",
         "exponential.mlir",
+        "iota.mlir",
         "log.mlir",
         "maximum.mlir",
         "minimum.mlir",
diff --git a/iree/test/e2e/xla_ops/CMakeLists.txt b/iree/test/e2e/xla_ops/CMakeLists.txt
index e2ab883..e4ae959 100644
--- a/iree/test/e2e/xla_ops/CMakeLists.txt
+++ b/iree/test/e2e/xla_ops/CMakeLists.txt
@@ -46,6 +46,7 @@
     "divide.mlir"
     "dot.mlir"
     "exponential.mlir"
+    "iota.mlir"
     "log.mlir"
     "maximum.mlir"
     "minimum.mlir"
@@ -92,6 +93,7 @@
     "divide.mlir"
     "dot.mlir"
     "exponential.mlir"
+    "iota.mlir"
     "log.mlir"
     "maximum.mlir"
     "minimum.mlir"