Updates mnist_tensorflow to use the vmla driver when running on CPU. Changes the `driver_name` variable from `"interpreter"` to `"vmla"` so that the notebook works when not using Vulkan. I also added `**/.ipynb_checkpoints/` to `.gitignore` to prevent that from accidentally being added. Closes https://github.com/google/iree/pull/1338 COPYBARA_INTEGRATE_REVIEW=https://github.com/google/iree/pull/1338 from google:phoenix-meadowlark-patch-1 bb461324d5a5a1d2a9c9d9af085c46cbb4bf4405 PiperOrigin-RevId: 304209468
diff --git a/.gitignore b/.gitignore index 990e1ad..c8b7550 100644 --- a/.gitignore +++ b/.gitignore
@@ -14,6 +14,7 @@ # Python *.pyc +**/.ipynb_checkpoints/ # Visual Studio files .vs/
diff --git a/colab/mnist_tensorflow.ipynb b/colab/mnist_tensorflow.ipynb index 590f36e..229bab3 100644 --- a/colab/mnist_tensorflow.ipynb +++ b/colab/mnist_tensorflow.ipynb
@@ -134,7 +134,7 @@ " driver_name = \"vulkan\"\n", "else:\n", " backend_name = \"vmla\"\n", - " driver_name = \"interpreter\"\n", + " driver_name = \"vmla\"\n", "tf.print(\"Using IREE compiler backend '%s' and runtime driver '%s'\" % (backend_name, driver_name))\n", "\n", "#@markdown -----\n", @@ -623,4 +623,4 @@ ] } ] -} \ No newline at end of file +}