Need to revise the BM kernel

GPU doesn’t like long nested for loops