[checkpoint] Address PR comments

Also add back SplitArgs pass in build_module.cc which somehow got lost in the shuffle.

(try again -- flaky test_crt.py test_autotune?)

src/relay/backend/build_module.cc

@@ -323,6 +323,12 @@ class RelayBuildModule : public runtime::ModuleNode {
         /*is_homogenous=*/config_->optional_homogeneous_target.defined(), /*is_vm=*/false);
     transform::PassContext pass_ctx = PassContext::Current();
 
+    if (config_->optional_homogeneous_target.defined()) {
+      // This pass currently only supports the homogeneous case.
+      pass_seqs.push_back(transform::SplitArgs(
+          config_->optional_homogeneous_target->GetAttr<Integer>("max_function_args", -1).value()));
+    }
+
     // Always plan devices so the remaining passes don't need to distinguish homogeneous vs
     // hetrogenous execution.
     pass_seqs.push_back(transform::PlanDevices(config_));

tests/python/unittest/test_micro_model_library_format.py

@@ -412,5 +412,4 @@ def test_export_byoc_c_module():
 if __name__ == "__main__":
     import sys
 
-    # sys.exit(pytest.main([__file__] + sys.argv[1:]))
-    test_export_operator_model_library_format()
+    sys.exit(pytest.main([__file__] + sys.argv[1:]))