Skip to content

pentschev/ucx-py

 
 

Repository files navigation

https://ucx-py.readthedocs.io/en/latest/

Python Bindings for UCX

Installing

Users can either install with Conda or build from source.

Testing

To run ucx-py's tests, just use pytest:

pytest -v

TCP Support

In order to use TCP add tcp to UCX_TLS and set UCXPY_IFNAME to the network interface you want to use. Some setup examples:

# TCP using "eth0" and CUDA support
export UCX_TLS=tcp,sockcm,cuda_copy,cuda_ipc
export UCX_SOCKADDR_TLS_PRIORITY=sockcm
export UCXPY_IFNAME="eth0"

# InfiniBand using "ib0" and CUDA support
export UCX_TLS=rc,sockcm,cuda_copy,cuda_ipc
export UCX_SOCKADDR_TLS_PRIORITY=sockcm
export UCXPY_IFNAME="ib0"

# TCP using "eno0" and no CUDA support
export UCX_TLS=tcp,sockcm
export UCX_SOCKADDR_TLS_PRIORITY=sockcm
export UCXPY_IFNAME="eno0"

About

Python bindings for UCX

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 65.4%
  • Cython 28.1%
  • Shell 3.9%
  • C 2.5%
  • Dockerfile 0.1%