DeepSpeed

9d07d756 - NameError: name 'mpu' is not defined (#305)

NameError: name 'mpu' is not defined (#305) mpu object is bound to the class instance.. the if statement uses `self.mpu' but just `mpu` is called in the following lines.. This raises a NameError