mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-11 17:00:22 +00:00
5d16ac958e
- configs.cuh: NUM_MAX_NVL_PEERS 8 -> 4 - internode.cu: introduce NvlPackT (uint64_t for 8 peers, uint32_t for 4) to handle packed-bool loads of is_token_in_rank; relax SourceMeta static_assert; replace 4 uint64_t-coupled sites - buffer.hpp/buffer.cc: relax NUM_MAX_NVL_PEERS assert (4 || 8); read MSCCLPP_EP_LOCAL_WORLD_SIZE env to override rdma_rank/nvl_rank partitioning when local world size != NUM_MAX_NVL_PEERS - CMakeLists.txt (ext/ep): rpath / install fix - pyproject.toml: MSCCLPP_BUILD_EXT_EP=ON - src/core/atomicadd_kernel.cu, kernels/buffer.cuh, kernels/utils.cuh: related EP fixes - test_internode_multirank.py: NUM_MAX_NVL_PEERS=4, rank %% 4
69 lines
1.5 KiB
TOML
69 lines
1.5 KiB
TOML
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT license.
|
|
|
|
[build-system]
|
|
requires = [
|
|
"scikit-build-core>=0.10.0",
|
|
"setuptools-scm[toml]>=8"
|
|
]
|
|
build-backend = "scikit_build_core.build"
|
|
|
|
[project]
|
|
name = "mscclpp"
|
|
dynamic = ["version"]
|
|
description = "MSCCL++ Python API"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"numpy",
|
|
"blake3",
|
|
"pybind11",
|
|
"sortedcontainers",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
cuda11 = ["cupy-cuda11x"]
|
|
cuda12 = ["cupy-cuda12x"]
|
|
cuda13 = ["cupy-cuda13x"]
|
|
rocm6 = ["cupy"]
|
|
benchmark = [
|
|
"mpi4py",
|
|
"prettytable",
|
|
"netifaces",
|
|
"matplotlib",
|
|
]
|
|
test = [
|
|
"pytest",
|
|
"mpi4py",
|
|
"netifaces",
|
|
]
|
|
|
|
[tool.setuptools_scm]
|
|
write_to = "python/mscclpp/_version.py"
|
|
version_scheme = "no-guess-dev"
|
|
|
|
[tool.scikit-build]
|
|
cmake.version = ">=3.25.0"
|
|
cmake.build-type = "Release"
|
|
build-dir = "build/{wheel_tag}"
|
|
# Tell scikit-build-core to get version from setuptools-scm
|
|
metadata.version.provider = "scikit_build_core.metadata.setuptools_scm"
|
|
|
|
[tool.scikit-build.wheel]
|
|
packages = ["python/mscclpp", "python/mscclpp_benchmark"]
|
|
install-dir = "mscclpp"
|
|
license-files = ["VERSION", "LICENSE", "CITATION.cff", "CODE_OF_CONDUCT.md", "README.md", "SECURITY.md", "SUPPORT.md"]
|
|
exclude = ["mscclpp/*.cpp"]
|
|
|
|
[tool.scikit-build.sdist]
|
|
include= ["python/mscclpp/_version.py"]
|
|
|
|
[tool.scikit-build.cmake.define]
|
|
MSCCLPP_BUILD_PYTHON_BINDINGS = "ON"
|
|
MSCCLPP_BUILD_TESTS = "OFF"
|
|
MSCCLPP_BUILD_EXT_EP = "ON"
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ['py310']
|
|
include = '\.pyi?$'
|