you check with dive company. We can only assume. I do not think you will be stamped out of Thailand, unless this is very Official Tour which requires you to enter Myanmar water and Immigration.
Thanks for the update, but what ever the reason is , in Kanchanaburi they will not stamp South African passport, glad to hear the at Mae Sai no restrictions in place.