reduce mem usage, removing b matrix, to add more elems in a, more parallel data

Signed-off-by: ankit <ankit@fire.(none)>