Device-Specific/GEMM : Fixed slice issue with GEMM