mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-07-14 19:27:20 +00:00
adc9ee5684
For mscclpp, to use nvls we require the buffer is allocated by mscclpp::GpuBuffer. Due to cupy doesn't support bfloat16 yet, we export the raw buffer to dlpack format. User can use this feature to create buffer with type supported by pytorch ```python buffer = RawGpuBuffer(1024 * 2) # 2 for bfloat16 dl_pack = buffer.to_dlpack(str(torch.bfloat16)) tensor = torch.utils.dlpack.from_dlpack(dl_pack) ```
29 lines
708 B
TOML
29 lines
708 B
TOML
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT license.
|
|
|
|
[build-system]
|
|
requires = ["scikit-build-core"]
|
|
build-backend = "scikit_build_core.build"
|
|
|
|
[project]
|
|
name = "mscclpp"
|
|
version = "0.6.0"
|
|
|
|
[tool.scikit-build]
|
|
cmake.version = ">=3.25.0"
|
|
cmake.build-type = "Release"
|
|
# for dlpack issue: https://github.com/microsoft/vcpkg/pull/44679
|
|
cmake.args = ["-DCMAKE_POLICY_VERSION_MINIMUM=3.5"]
|
|
build-dir = "build/{wheel_tag}"
|
|
wheel.packages = ["python/mscclpp", "python/mscclpp_benchmark"]
|
|
wheel.install-dir = "mscclpp"
|
|
|
|
[tool.scikit-build.cmake.define]
|
|
MSCCLPP_BUILD_PYTHON_BINDINGS = "ON"
|
|
MSCCLPP_BUILD_TESTS = "OFF"
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ['py38']
|
|
include = '\.pyi?$'
|