DeepWok / mase

Machine-Learning Accelerator System Exploration Tools
Other
117 stars 52 forks source link

flop_estimator module is broken. #54

Closed btarifi10 closed 2 months ago

btarifi10 commented 7 months ago

The flop_estimator module cannot be loaded due to some faulty imports, such as get_input_args in deepspeed.py.

I have attempted to fix this but it is ambiguous on what exactly is required and time is limited.