[checkpoint] TODO's for default device logic

apache · Oct 2, 2021 · 3c917f0 · 3c917f0
1 parent 9dc0882
commit 3c917f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/relay/backend/build_module.cc b/src/relay/backend/build_module.cc
@@ -332,6 +332,7 @@ class RelayBuildModule : public runtime::ModuleNode {
     Array<Pass> pass_seqs = GetPassPrefix(targets_, false);
     transform::PassContext pass_ctx = PassContext::Current();
 
+    // TODO(mbs): Centralize this logic and reconcile with similar in relay/backend/vm/compiler.cc
     DLDeviceType default_device_type;
     if (targets_.size() == 1) {
       // Homogenous execution.

diff --git a/src/relay/backend/vm/compiler.cc b/src/relay/backend/vm/compiler.cc
@@ -1031,6 +1031,7 @@ IRModule VMCompiler::OptimizeModule(IRModule mod, const TargetsMap& targets_arg,
 
   Array<Pass> pass_seqs = relay::backend::GetPassPrefix(targets, true);
 
+  // TODO(mbs): Reconcile with relay/backend/build_module.cc
   DLDeviceType default_device_type;
   if (targets_arg.size() == 1) {
     default_device_type =