got basic tgemm_amm working