Update on "Add a centralized TORCH_LIBRARY declaration for quantized and xnnpack ops."

ezyang · ezyang · commit 762b42b6f137 · 2020-04-15T11:07:23.000-07:00
Fixes #36510 Note that this introduces some schema string duplication, but I'll kill the dupes when I migrate the individual files to the new registration API. Signed-off-by: Edward Z. Yang <ezyang@fb.com> Differential Revision: [D21005369](https://our.internmc.facebook.com/intern/diff/D21005369) [ghstack-poisoned]
diff --git a/aten/src/ATen/core/op_registration/op_registration.cpp b/aten/src/ATen/core/op_registration/op_registration.cpp
@@ -20,10 +20,13 @@ namespace {
     switch (kind) {
       case Library::DEF:
         os << "TORCH_LIBRARY";
+        break;
       case Library::IMPL:
         os << "TORCH_LIBRARY_IMPL";
+        break;
       case Library::FRAGMENT:
         os << "TORCH_LIBRARY_FRAGMENT";
+        break;
     }
     return os;
   }

Original file line number	Diff line number	Diff line change
`@@ -20,10 +20,13 @@ namespace {`
`20`	`20`	`switch (kind) {`
`21`	`21`	`case Library::DEF:`
`22`	`22`	`os << "TORCH_LIBRARY";`
	`23`	`+ break;`
`23`	`24`	`case Library::IMPL:`
`24`	`25`	`os << "TORCH_LIBRARY_IMPL";`
	`26`	`+ break;`
`25`	`27`	`case Library::FRAGMENT:`
`26`	`28`	`os << "TORCH_LIBRARY_FRAGMENT";`
	`29`	`+ break;`
`27`	`30`	`}`
`28`	`31`	`return os;`
`29`	`32`	`}`