提交 d70668cb 编写于 作者: K kuizhiqing 提交者: Wei Shengyu

fix typo

上级 a37cee70
......@@ -86,8 +86,7 @@ class RollWrapperSingleton():
def __init__(self):
self.index_dict = {}
self.roll_with_index_select = RollWithIndexSelect.apply
if 'npu' in paddle.device.get_all_custom_device_type():
self.enable = True
self.enable = True if 'npu' in paddle.device.get_all_custom_device_type() else False
def __call__(self, x, shifts, axis):
if not self.enable:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册