This is the pybind for the AMGPCG solver in yuchen-sun-cg/lfm using pytorch extension. A wrapper for warp is also provided in amgpcg_pybind/warp.py.
This code is used by LFM, VPFM.
To install, run:
pip install .Make sure that you always import torch first before import amgpcg_pybind.