commit 221049c9fd1622aa6d26ab2500b76e0b151be5db
author Han-Chung Wang <hanhanW@users.noreply.github.com> Thu Jul 23 10:32:40 2020 -0700
committer GitHub <noreply@github.com> Thu Jul 23 10:32:40 2020 -0700
tree ee0a8365f15a3d6f5d44e6b6f2a42e35f287360d
parent f83d66b5d3720dfda23e61999a3b1371b89a5d18

Add iota.mlir test to e2e/xla_ops/ (#2630)

This op is supported in VMLA but doesn't have a test.

iree/test/e2e/xla_ops/iota.mlir

diff --git a/iree/test/e2e/xla_ops/iota.mlir b/iree/test/e2e/xla_ops/iota.mlir
new file mode 100644
index 0000000..c3ff040
--- /dev/null
+++ b/iree/test/e2e/xla_ops/iota.mlir
@@ -0,0 +1,16 @@
+func @iota_dim0() attributes { iree.module.export } {
+  %result = "mhlo.iota"() {iota_dimension = 0 : i64} : () -> tensor<2x3xf32>
+  check.expect_almost_eq_const(%result, dense<[
+    [0.0, 0.0, 0.0],
+    [1.0, 1.0, 1.0]]> : tensor<2x3xf32>) : tensor<2x3xf32>
+  return
+}
+
+
+func @iota_dim1() attributes { iree.module.export } {
+  %result = "mhlo.iota"() {iota_dimension = 1 : i64} : () -> tensor<2x3xf32>
+  check.expect_almost_eq_const(%result, dense<[
+    [0.0, 1.0, 2.0],
+    [0.0, 1.0, 2.0]]> : tensor<2x3xf32>) : tensor<2x3xf32>
+  return
+}