make failing with torch_isolation/CMakeFiles/torchrot.dir/src/rotations.cpp.o] Error 1

We are trying to build polymetis on aarch64 system.

Steps:

git clone git@github.com:facebookresearch/fairo

cmake .. -DCMAKE_BUILD_TYPE=Release -DBUILD_FRANKA=OFF -DBUILD_TESTS=ON -DBUILD_DOCS=ON -DCMAKE_PREFIX_PATH=$CONDA_PREFIX -DBoost_NO_SYSTEM_PATHS=ON -DBOOST_ROOT=$CONDA_PREFIX -DBoost_INCLUDE_DIR=$CONDA_PREFIX/include -DBoost_LIBRARY_DIR=$CONDA_PREFIX/lib -DBoost_PYTHON_LIBRARY=$CONDA_PREFIX/lib/libboost_python310.so -DPYTHON_EXECUTABLE=$(which python)

make -j

Fails with the following error:

97 | FanModeType mode = torch::kFanIn,
| ~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
/proj/work/arunimag/Anaconda/envs/sphinx_env_20/lib/python3.10/site-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h: In instantiation of ‘void torch::nn::ConvNdImpl::reset_parameters() [with long unsigned int D = 1; Derived = torch::nn::Conv1dImpl]’:
/proj/work/arunimag/Anaconda/envs/sphinx_env_20/lib/python3.10/site-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:99:5: required from ‘void torch::nn::ConvNdImpl::reset() [with long unsigned int D = 1; Derived = torch::nn::Conv1dImpl]’
/proj/work/arunimag/Anaconda/envs/sphinx_env_20/lib/python3.10/site-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:31:8: required from here
/proj/work/arunimag/Anaconda/envs/sphinx_env_20/lib/python3.10/site-packages/torch/include/torch/csrc/api/include/torch/nn/modules/conv.h:103:27: error: cannot convert ‘const torch::enumtype::kFanIn’ to ‘int’
103 | init::kaiming_uniform_(
| ~~~~~~~~~~~~~~~~~~~~~~^
| |
| const torch::enumtype::kFanIn
104 | weight,
| ~~~~~~~
105 | /*a=*/std::sqrt(5)); // NOLINT(cppcoreguidelines-avoid-magic-numbers)
| ~~~~~~~~~~~~~~~~~~~
/proj/work/arunimag/Anaconda/envs/sphinx_env_20/lib/python3.10/site-packages/torch/include/torch/csrc/api/include/torch/nn/init.h:97:17: note: initializing argument 3 of ‘at::Tensor torch::nn::init::kaiming_uniform_(at::Tensor, double, int, int)’
97 | FanModeType mode = torch::kFanIn,
| ~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~
make[2]: *** [torch_isolation/CMakeFiles/torchrot.dir/build.make:76: torch_isolation/CMakeFiles/torchrot.dir/src/rotations.cpp.o] Error 1
make[1]: *** [CMakeFiles/Makefile2:379: torch_isolation/CMakeFiles/torchrot.dir/all] Error 2
make: *** [Makefile:136: all] Error 2