loop_parallel_ast.py 文件源码

def visit_Subscript(self, node):

        ind_node = node.slice
        for field, value in ast.iter_fields(ind_node):
            if field == 'value':
                if isinstance(value, ast.Tuple):
                    try:
                        for i in range(len(value.elts)):
                            ind_value = py_ast.dump_ast(value.elts[i])
                            self.value.append(ind_value)
                            self.write.append([])
                            self.visit(value.elts[i])
                    except:
                        pass
                else:
                    try:
                        ind_value = py_ast.dump_ast(value)
                        self.value.append(ind_value)
                        self.write.append([])
                        self.visit(value)
                    except:
                        pass

            elif field == 'dims':
                try:
                    for i in range(len(value)):
                        ind_value = py_ast.dump_ast(value[i])
                        self.value.append(ind_value)
                        self.write.append([])
                        self.visit(value[i])
                except:
                    pass