Error: Core dumped using -beagle_gpu option with beast

[ozankiratli]

When I run beagle with gpu option I get this message:

java: /root/beagle-lib/libhmsbeagle/GPU/BeagleGPUImpl.hpp:963: int beagle::gpu::cuda::BeagleGPUImpl<Real>::setPartials(int, const double*) [with Real = double]: Assertion `kLastTipPartialsBufferIndex >= 0 && kLastTipPartialsBufferIndex < kTipPartialsBufferCount' failed.
Aborted (core dumped)

Running with cpu has no issues.

OS: Ubuntu 20.04
NVIDIA: GeForce GTX 1050
JAVA: openjdk 11.0.7 2020-04-14

I set beagle-lib manually. The setting LD_LIBRARY_PATH didn't work for me. I've created symbolic links to the library files. Here's how I set it. I've found where nvcc is with which nvcc

git clone --depth=1 https://github.com/beagle-dev/beagle-lib.git
cd beagle-lib/
./configure --prefix=/opt/beagle-lib/ --with-cuda=/usr/bin/
make install
cd /opt/beagle-lib
for i in libhmsbeagle* ; do ln -s /opt/beagle-lib/lib/$i /usr/lib/$i ; done

make check returns no errors.

When I run beast-mcmc -beagle_info I get the following:

  BEAST v1.10.4 Prerelease #bc6cbd9, 2002-2018
       Bayesian Evolutionary Analysis Sampling Trees
                 Designed and developed by
   Alexei J. Drummond, Andrew Rambaut and Marc A. Suchard
                              
               Department of Computer Science
                   University of Auckland
                  [email protected]
                              
             Institute of Evolutionary Biology
                  University of Edinburgh
                     [email protected]
                              
              David Geffen School of Medicine
           University of California, Los Angeles
                     [email protected]
                              
                Downloads, Help & Resources:
                  	http://beast.community
                              
Source code distributed under the GNU Lesser General Public License:
          	http://github.com/beast-dev/beast-mcmc
                              
                     BEAST developers:
	Alex Alekseyenko, Guy Baele, Trevor Bedford, Filip Bielejec, Erik Bloomquist, Matthew Hall,
	Joseph Heled, Sebastian Hoehna, Denise Kuehnert, Philippe Lemey, Wai Lok Sibon Li,
	Gerton Lunter, Sidney Markowitz, Vladimir Minin, Michael Defoin Platel,
          	Oliver Pybus, Chieh-Hsi Wu, Walter Xie
                              
                         Thanks to:
    	Roald Forsberg, Beth Shapiro and Korbinian Strimmer


Using BEAGLE library v3.2.0 (PRE-RELEASE) for accelerated, parallel likelihood evaluation
2009-, BEAGLE Working Group - https://beagle-dev.github.io/
Citation: Ayres et al (2012) Systematic Biology 61: 170-173 | doi:10.1093/sysbio/syr100


BEAGLE resources available:
0 : CPU
    Flags: PRECISION_SINGLE PRECISION_DOUBLE COMPUTATION_SYNCH EIGEN_REAL EIGEN_COMPLEX SCALING_MANUAL SCALING_AUTO SCALING_ALWAYS SCALERS_RAW SCALERS_LOG VECTOR_SSE VECTOR_NONE THREADING_CPP THREADING_NONE PROCESSOR_CPU FRAMEWORK_CPU


1 : GeForce GTX 1050
    Global memory (MB): 4042
    Clock speed (Ghz): 1.49
    Number of cores: 640
    Flags: PRECISION_SINGLE PRECISION_DOUBLE COMPUTATION_SYNCH COMPUTATION_ASYNCH EIGEN_REAL EIGEN_COMPLEX SCALING_MANUAL SCALING_AUTO SCALING_ALWAYS SCALERS_RAW SCALERS_LOG VECTOR_NONE THREADING_NONE PROCESSOR_GPU FRAMEWORK_CUDA PARALLELOPS_STREAMS PARALLELOPS_GRID