[go: up one dir, main page]

Skip to content

cvsik/pybtas

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pybtas: A python interface to BTAS

We provide an interface to CP3 which takes a 3-tensor with symmetric values under exchange of the first two tensor indices. We return the factorized matrices which can be used in THC.

Installation

you will need cmake and pybind

conda install cmake python -m pip install pybind

  1. get boost and eigen and btas and blas/lapack. I used OpenBLAS which comes with lapack.
  2. build boost. navigate to boost dir and type ./bootstrap.sh && ./b2
  3. navigate to btas and use

cmake . -DBTAS_USE_BLAS_LAPACK=ON -DCMAKE_INSTALL_PREFIX=~/opt/btas -DBTAS_BUILD_UNITTEST=TRUE -DBTAS_ASSERT_THROWS=ON DBOOST_INCLUDEDIR=/usr/local/google/home/nickrubin/opt/boost_1_76_0 -DBoost_DIR=/usr/local/google/home/nickrubin/opt/boost_1_76_0

or the equivalent for your system

then type

make check

and installation should happen along with checks. Conclude by typing

make install

  1. navigate to pybtas directory and update CMakeLists.txt with with appropriate paths for eigen and BTAS

  2. python setup.py clean && python setup.py install

Then run test_pybtas.py to run CP3.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 68.3%
  • Python 25.1%
  • CMake 6.5%
  • Other 0.1%