Remove strip_overloads from native_training example.
This is no longer necessary, and the symbol name was changed recently,
causing the build to break:
https://buildkite.com/iree/iree-samples/builds/563#0184e9f1-f3e6-415f-81d7-d9b8d088cb25
diff --git a/samples/native_training/native_training.py b/samples/native_training/native_training.py
index 2e17e82..3020a7b 100644
--- a/samples/native_training/native_training.py
+++ b/samples/native_training/native_training.py
@@ -8,7 +8,6 @@
import os
import functorch
-from functorch._src.compile_utils import strip_overloads
import iree_torch
import torch
import torch_mlir
@@ -85,10 +84,6 @@
train_args = (w, b, X_test, y_test)
graph = functorch.make_fx(train)(*train_args)
- # TODO: Remove once https://github.com/llvm/torch-mlir/issues/1495
- # is resolved.
- strip_overloads(graph)
-
mlir = torch_mlir.compile(graph,
train_args,
output_type=torch_mlir.OutputType.LINALG_ON_TENSORS)
