Test failure: Gather Forward and Backward Kernel

I installed FlexFlow `master` from https://github.com/flexflow/flexflow-train/blob/master/docs/SAPLING.md on Sapling. CPU tests pass, but GPU tests fail with: ```console $ proj test ----FAILED kernels:Test Gather Forward and Backward Kernel---------------------- ----To debug, run: proj test --debug 'kernels:Test Gather Forward and Backward Kernel' STDOUT: =============================================================================== ../../../../../lib/kernels/test/src/test_gather_kernels.cc:8: TEST SUITE: cuda-kernels-tests TEST CASE: Test Gather Forward and Backward Kernel gpu_backward_kernel test gather backward, 2D ../../../../../lib/kernels/test/src/test_gather_kernels.cc:99: ERROR: CHECK( contains_non_zero(input_grad_accessor) ) is NOT correct! values: CHECK( false ) =============================================================================== [doctest] test cases: 1 | 0 passed | 1 failed | 56 skipped [doctest] assertions: 3 | 2 passed | 1 failed | [doctest] Status: FAILURE! STDERR: Test results: 641 passed / 1 failed / 642 total ERROR:proj.failure:The following tests failed: - kernels:Test Gather Forward and Backward Kernel ``` Using FlexFlow commit 5abb22548b5dcfa0f7eee6ca4421d5cd8472a2ac