Add Autograd Fallback for MTIA (pytorch#139211)

egienvalue · pytorchmergebot · commit 1a701853098b · 2024-11-05T20:58:21.000Z
Summary: As title. Test Plan: OSS and internal CIs. Differential Revision: D65022481 Pull Request resolved: pytorch#139211 Approved by: https://github.com/jvandebon
diff --git a/aten/src/ATen/core/VariableFallbackKernel.cpp b/aten/src/ATen/core/VariableFallbackKernel.cpp
@@ -76,6 +76,10 @@ TORCH_LIBRARY_IMPL(_, AutogradCUDA, m) {
   m.fallback(AUTOGRAD_FALLBACK);
 }
 
+TORCH_LIBRARY_IMPL(_, AutogradMTIA, m) {
+  m.fallback(AUTOGRAD_FALLBACK);
+}
+
 TORCH_LIBRARY_IMPL(_, AutogradXLA, m) {
   m.fallback(AUTOGRAD_FALLBACK);
 }

Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,10 @@ TORCH_LIBRARY_IMPL(_, AutogradCUDA, m) {`
`76`	`76`	`m.fallback(AUTOGRAD_FALLBACK);`
`77`	`77`	`}`
`78`	`78`
	`79`	`+TORCH_LIBRARY_IMPL(_, AutogradMTIA, m) {`
	`80`	`+ m.fallback(AUTOGRAD_FALLBACK);`
	`81`	`+}`
	`82`	`+`
`79`	`83`	`TORCH_LIBRARY_IMPL(_, AutogradXLA, m) {`
`80`	`84`	`m.fallback(AUTOGRAD_FALLBACK);`
`81`	`85`	`}`