Minor fixes
Add trsm kernel
Fix kernels