In order to be compatible with iree-turbine, make iree-turbine can training#13

Open

mollon650 wants to merge 1 commit intoiree-org:mainfrom mollon650:origin

+145-21

Commits on May 10, 2024

[iree-turbine] support simple mlp training for cuda by dynamo & ignore torch.none when it appear in backward graph
mollon650
committed