support patch ascend
This commit is contained in:
2
patch.ascend/device.py.patch
Normal file
2
patch.ascend/device.py.patch
Normal file
@@ -0,0 +1,2 @@
|
||||
27d26
|
||||
< assert eles[0] in ['cpu', 'cuda', 'gpu'], err_msg
|
||||
Reference in New Issue
Block a user