mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-16 02:54:48 +00:00
5acac93dbc
Provides two integration ways for MSCCL++ DSL. 1. Integrate with customized communication group 2. Integrate with NCCL API Introduce new Python APIs to make it work: ```python mscclpp.compile # compile dsl to json based execution plan mscclpp.ExecutionPlanRegistry.register_plan(plan) # register the compiled plan to executionPlanRegistery mscclpp.ExecutionPlanRegistry.set_selector(selector) # set the selector, the selector will return the best execution plan based on collection, message size, world size.... ``` Fix #556 --------- Co-authored-by: Caio Rocha <caiorocha@microsoft.com> Co-authored-by: Changho Hwang <changhohwang@microsoft.com>
16 lines
496 B
CMake
16 lines
496 B
CMake
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT license.
|
|
|
|
add_subdirectory(csrc)
|
|
add_subdirectory(test)
|
|
|
|
add_custom_target(pytest_lib_copy ALL
|
|
COMMAND ${CMAKE_COMMAND} -E copy_if_different
|
|
${CMAKE_CURRENT_BINARY_DIR}/csrc/_mscclpp.*.so
|
|
${CMAKE_CURRENT_SOURCE_DIR}/mscclpp
|
|
COMMAND ${CMAKE_COMMAND} -E copy_if_different
|
|
${CMAKE_CURRENT_BINARY_DIR}/test/_ext.*.so
|
|
${CMAKE_CURRENT_SOURCE_DIR}/test/_cpp
|
|
DEPENDS mscclpp_py mscclpp_py_test
|
|
)
|