gpu code with warp partitioning