The actor worked hard to get in shape for the underwater action sequences in his upcoming film