From 227dbb87260b2e14d030a6d880c4f69d475c7022 Mon Sep 17 00:00:00 2001 From: Yong Wu Date: Tue, 14 May 2024 22:28:32 -0700 Subject: [PATCH] Add false for arg worker0_only in disco.empty (#2344) --- cpp/llm_chat.cc | 2 +- cpp/serve/function_table.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cpp/llm_chat.cc b/cpp/llm_chat.cc index 93de185eb2..a8d2edc11a 100644 --- 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 index d63857d539..2ed864f298 100644 --- 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); }