Wmma 3 mods working for wmma 4