Commit 4017bd18 authored by Neelay Shah's avatar Neelay Shah Committed by GitHub
Browse files

refactor: worker rename to runtime

parent 8492333b
...@@ -24,7 +24,7 @@ from llm.vllm.operators.vllm import ( ...@@ -24,7 +24,7 @@ from llm.vllm.operators.vllm import (
VllmOperator, VllmOperator,
) )
from triton_distributed.worker import Deployment, OperatorConfig, WorkerConfig from triton_distributed.runtime import Deployment, OperatorConfig, WorkerConfig
from .parser import parse_args from .parser import parse_args
......
...@@ -8,7 +8,7 @@ import numpy as np ...@@ -8,7 +8,7 @@ import numpy as np
from triton_distributed.icp.data_plane import DataPlane from triton_distributed.icp.data_plane import DataPlane
from triton_distributed.icp.request_plane import RequestPlane from triton_distributed.icp.request_plane import RequestPlane
from triton_distributed.worker import ( from triton_distributed.runtime import (
Operator, Operator,
RemoteInferenceRequest, RemoteInferenceRequest,
RemoteInferenceResponse, RemoteInferenceResponse,
......
...@@ -12,20 +12,20 @@ ...@@ -12,20 +12,20 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from triton_distributed.worker.deployment import Deployment as Deployment from triton_distributed.runtime.deployment import Deployment as Deployment
from triton_distributed.worker.logger import get_logger as get_logger from triton_distributed.runtime.logger import get_logger as get_logger
from triton_distributed.worker.logger import get_logger_config as get_logger_config from triton_distributed.runtime.logger import get_logger_config as get_logger_config
from triton_distributed.worker.operator import Operator as Operator from triton_distributed.runtime.operator import Operator as Operator
from triton_distributed.worker.operator import OperatorConfig as OperatorConfig from triton_distributed.runtime.operator import OperatorConfig as OperatorConfig
from triton_distributed.worker.remote_operator import RemoteOperator as RemoteOperator from triton_distributed.runtime.remote_operator import RemoteOperator as RemoteOperator
from triton_distributed.worker.remote_request import ( from triton_distributed.runtime.remote_request import (
RemoteInferenceRequest as RemoteInferenceRequest, RemoteInferenceRequest as RemoteInferenceRequest,
) )
from triton_distributed.worker.remote_response import ( from triton_distributed.runtime.remote_response import (
RemoteInferenceResponse as RemoteInferenceResponse, RemoteInferenceResponse as RemoteInferenceResponse,
) )
from triton_distributed.worker.triton_core_operator import ( from triton_distributed.runtime.triton_core_operator import (
TritonCoreOperator as TritonCoreOperator, TritonCoreOperator as TritonCoreOperator,
) )
from triton_distributed.worker.worker import Worker as Worker from triton_distributed.runtime.worker import Worker as Worker
from triton_distributed.worker.worker import WorkerConfig as WorkerConfig from triton_distributed.runtime.worker import WorkerConfig as WorkerConfig
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment