/VTensor

VTensor, a C++ library, facilitates tensor manipulation on GPUs, emulating the python-numpy style for ease of use. It leverages RMM (RAPIDS Memory Manager) for efficient device memory management. It also supports xtensor for host memory operations.

Primary LanguageC++Apache License 2.0Apache-2.0

VTensor

VTensor, a C++ library, facilitates tensor manipulation on GPUs, emulating the python-numpy style for ease of use. It leverages RMM (RAPIDS Memory Manager) for efficient device memory management. The library integrates CuRand, CuBLAS, and CuSolver to support a wide range of operations, including mathematics, linear algebra, and random number generation. Please visit our website https://vorticity-inc.github.io/VTensor/ for more information!

Test

bazel test ...

Formatter

find lib  -name '*.hpp' |  xargs clang-format -i -style=file:clang.yaml

Update docs

doxygen docs/Doxyfile.in
sphinx-build -b html docs/source docs/build/html

Future updates

  • Support Solvers (Cholesky, QR etc.)
  • Support GPUDirect
  • Support more matrix operations
  • Support Sparse martix with CuSparse