Add false for arg worker0_only in disco.empty (#2344)

mlc-ai · May 15, 2024 · 227dbb8 · 227dbb8
1 parent 2bbbd52
commit 227dbb8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cpp/llm_chat.cc b/cpp/llm_chat.cc
@@ -302,7 +302,7 @@ struct FunctionTable {
     Device null_device{DLDeviceType(0), 0};
     if (this->use_disco) {
       DRef empty_func = sess->GetGlobalFunc("runtime.disco.empty");
-      return sess->CallPacked(empty_func, shape, dtype, null_device);
+      return sess->CallPacked(empty_func, shape, dtype, null_device, false);
     } else {
       return NDArray::Empty(shape, dtype, device);
     }

diff --git a/cpp/serve/function_table.cc b/cpp/serve/function_table.cc
@@ -270,7 +270,7 @@ ObjectRef FunctionTable::Empty(ShapeTuple shape, DataType dtype, Device device)
   Device null_device{DLDeviceType(0), 0};
   if (this->use_disco) {
     DRef empty_func = sess->GetGlobalFunc("runtime.disco.empty");
-    return sess->CallPacked(empty_func, shape, dtype, null_device);
+    return sess->CallPacked(empty_func, shape, dtype, null_device, false);
   } else {
     return NDArray::Empty(shape, dtype, device);
   }