Looks like we need to make a small change to the header to add a new template parameter as a follow up to https://github.com/rapidsai/raft/issues/2832