User guide

Contents

• Benchmark results
  • Flame graph of original implementation
  • Flame graph of CPU implementation
  • Flame graph of GPU implementation
  • Random (N, N, N, N) tensor
• API Reference
  • legendre_decomp

Installation

Installing CuPy

(Recommended) install one of the binary wheels depending on the system’s installed CUDA Toolkit version.

$ pip install cupy-cuda11x  # CUDA 11.x
$ pip install cupy-cuda12x  # CUDA 12.x
$ pip install cupy-rocm-4-3  # AMD ROCm 4.3
$ pip install cupy-rocm-5-0  # AMD ROCm 5.0

Other options for installing CuPy can be found here

Installing LegendreDecomposition

Install using either poetry or pip.

Poetry

$ poetry install

Pip

$ pip install [-e] .

Project structure

• legendredecomposition/: Main module

• docs/: Documentation

• tests/: Functional tests

• scripts/: Benchmark scripts

• notebooks/: Benchmark Jupyter Notebooks

Indices and tables

• Index

• Module Index

• Search Page