From a0359c7510e8ce03aea066f8bf91d4557abc1698 Mon Sep 17 00:00:00 2001 From: Jae-Won Chung Date: Fri, 13 Oct 2023 11:25:09 -0400 Subject: [PATCH] Fix doc build --- mkdocs.yml | 3 +++ zeus/optimizer/perseus/server/__init__.py | 22 ++++++++++++++++++++++ zeus/util/framework.py | 3 --- 3 files changed, 25 insertions(+), 3 deletions(-) create mode 100644 zeus/optimizer/perseus/server/__init__.py diff --git a/mkdocs.yml b/mkdocs.yml index d06f10f2..20d60caa 100644 --- a/mkdocs.yml +++ b/mkdocs.yml @@ -111,6 +111,9 @@ nav: - getting_started/index.md - Environment Setup: getting_started/environment.md - Installing and Building: getting_started/installing_and_building.md + - Perseus: + - perseus/index.md + - Integrating: perseus/integrating.md - Extending Zeus: extend.md - Source Code Reference: reference/ diff --git a/zeus/optimizer/perseus/server/__init__.py b/zeus/optimizer/perseus/server/__init__.py new file mode 100644 index 00000000..3331ddaa --- /dev/null +++ b/zeus/optimizer/perseus/server/__init__.py @@ -0,0 +1,22 @@ +# Copyright (C) 2023 Jae-Won Chung +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +"""The Perseus server guides the PerseusOptimizer with frequency plans. + +The server is agnostic to the training framework the PerseusOptimizer +is integrated with. A server is useful because large model training is +typically distributed, and we still need one place to coordinate the +frequency plans. Later, the server will be extended to support complete +online profiling and optimization. +""" diff --git a/zeus/util/framework.py b/zeus/util/framework.py index 02578d1d..74653ea1 100644 --- a/zeus/util/framework.py +++ b/zeus/util/framework.py @@ -57,6 +57,3 @@ def cuda_sync(device: int | None = None) -> None: return raise RuntimeError("No frameworks are available.") - - -torch_is_available()