Skip to content

Commit

Permalink
v0.3.0 (#139)
Browse files Browse the repository at this point in the history
  • Loading branch information
chhwang authored Oct 4, 2023
1 parent b2b7a8b commit 6cb4b6e
Show file tree
Hide file tree
Showing 6 changed files with 21 additions and 32 deletions.
2 changes: 1 addition & 1 deletion CITATION.cff
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
cff-version: 1.2.0
title: "ARK: A GPU-driven system framework for scalable AI applications"
version: 0.2.1
version: 0.3.0
message: >-
If you use this project in your research, please cite it as below.
authors:
Expand Down
4 changes: 2 additions & 2 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
# Licensed under the MIT license.

set(ARK_MAJOR "0")
set(ARK_MINOR "2")
set(ARK_PATCH "1")
set(ARK_MINOR "3")
set(ARK_PATCH "0")

set(ARK_VERSION "${ARK_MAJOR}.${ARK_MINOR}.${ARK_PATCH}")
set(ARK_SOVERSION "${ARK_MAJOR}.${ARK_MINOR}")
Expand Down
37 changes: 13 additions & 24 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,35 +21,24 @@ ARK provides a set of APIs for users to express their distributed deep learning

ARK is under active development and a part of its features will be added in a future release. The following describes key features of each version.

### New in ARK v0.2 (Latest Release)
### New in ARK v0.3 (Latest Release)

* Change the `Tensor` interface to expose the underlying buffer
* Extend communication interfaces to support more communication operators
* Add many new operators
* Upgrade CUTLASS to v3.2.0
* Change the Python `Runtime.run()` interface
* Add a new parallel matmul example
* Update the documentation
* Improve unit tests
* Add an Azure pipeline for unit tests
* Add code coverage
* Many bug fixes
* Enable heuristic model graph optimization
* Revise Python interfaces
* Add more operators & support mixed-precision models & support `bfloat16`
* Add a Llama2-7B example
* Fix connection setup bugs for large & distributed models
* Fix correctness bugs from a few operators
* Minor scheduler improvements

See details from https://github.com/microsoft/ark/issues/61.

### ARK v0.3 (TBU, Oct. 2023)

* A simple operator scheduler for debugging
* Simpler communication interfaces
* Support more operators
* More inference examples
See details from https://github.com/microsoft/ark/issues/113.

### ARK v0.4 (TBU, Nov. 2023)

* Full support for the operator profiler
* High-performance collective communication
* Support more operators
* More inference & training examples
* Support AMD GPUs
* Add high-performance AllReduce & AllGather algorithms
* Add multi-GPU LLM examples
* Improve Python unit tests & code coverage

## Contributing

Expand Down
4 changes: 2 additions & 2 deletions ark/include/ark.h
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,8 @@
#include <vector>

#define ARK_MAJOR 0
#define ARK_MINOR 2
#define ARK_PATCH 1
#define ARK_MINOR 3
#define ARK_PATCH 0
#define ARK_VERSION (ARK_MAJOR * 10000 + ARK_MINOR * 100 + ARK_PATCH)

namespace ark {
Expand Down
4 changes: 2 additions & 2 deletions docs/sphinx/source/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@
project = "ARK"
copyright = "2023, ARK Team"
author = "ARK Team"
version = "0.2.1"
release = "0.2.1"
version = "0.3.0"
release = "0.3.0"

# -- General configuration ---------------------------------------------------

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ build-backend = "scikit_build_core.build"

[project]
name = "ark"
version = "0.2.1"
version = "0.3.0"

[tool.scikit-build]
cmake.minimum-version = "3.25"
Expand Down

0 comments on commit 6cb4b6e

Please sign in to comment.