Add helper functions for setting up Nbnxm gpu object in nblib