This repository was archived by the owner on Jan 26, 2026. It is now read-only.
File tree Expand file tree Collapse file tree 1 file changed +29
-0
lines changed
Expand file tree Collapse file tree 1 file changed +29
-0
lines changed Original file line number Diff line number Diff line change 1+ # Distributed Data-Parallel Python Tensor
2+ A tensor implementation following the [ array API as defined by the data-API consortium] ( https://data-apis.org/array-api/latest/index.html ) .
3+ It supports a controller-worker execution model as well as a CSP-like execution.
4+
5+ ## Setting up build environment
6+ ``` bash
7+ conda env create -f conda-env.yml -n ddpt
8+ conda activate ddpt
9+ export MPIROOT=$CONDA_PREFIX
10+ export MKLROOT=$CONDA_PREFIX
11+ git --recurse-submodules clone https://github.com/intel-sandbox/personal.fschlimb.ddptensor
12+ ```
13+ ## Building ddptensor
14+ ``` bash
15+ cd personal.fschlimb.ddptensor
16+ CC=gcc-9 CXX=g++-9 CMAKE_BUILD_PARALLEL_LEVEL=8 python setup.py develop
17+ ```
18+
19+ ## Running Tests
20+ ``` bash
21+ # single rank
22+ pytest test
23+ # multiple ranks, controller-worker, controller spawns ranks
24+ DDPT_MPI_SPAWN=$NoW PYTHON_EXE=` which python` pytest test
25+ # multiple ranks, controller-worker, mpirun
26+ mpirun -n $N python -m pytest test
27+ # multiple ranks, CSP
28+ DDPT_CW=0 mpirun -n $N python -m pytest test
29+ ```
You can’t perform that action at this time.
0 commit comments