Add batch support to blocksparse.