update version

Author: a710128

setup.py

@@ -73,7 +73,7 @@ def get_device_cc():
 
 setup(
     name='bmtrain',
-    version='0.2.0',
+    version='0.2.1',
     author="Guoyang Zeng",
     author_email="qbjooo@qq.com",
     description="A toolkit for training big models",