def create_subr_read_typedecl_in_module(self, node): parent = node.kgen_parent stmt = node.kgen_stmt raw_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state(stmt.geninfo) ]) entity_names = [ e for e in raw_entity_names if not stmt.get_variable(e).is_parameter() ] raw_out_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state_out(stmt.geninfo) ]) out_entity_names = [ e for e in raw_out_entity_names if not stmt.get_variable(e).is_parameter() ] def get_attrs(attrspec, allowed_attrs): attrspec = [] for attr in stmt.attrspec: if any( attr.startswith(allowed_attr) for allowed_attr in allowed_attrs): attrspec.append(attr) return attrspec def get_decls(names, decls, prefix=''): import re entity_decls = [] for decl in decls: ename = re.split('\(|\*|=', decl)[0].strip() if ename in names: entity_decls.append(prefix + decl) return entity_decls if len(entity_names) == 0: node.kgen_forced_line = False elif len(entity_names) != len(stmt.entity_decls): attrspec = get_attrs( stmt.attrspec, ['pointer', 'allocatable', 'dimension', 'public', 'target']) entity_decls = get_decls(entity_names, stmt.entity_decls) attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} if stmt.is_derived(): node.type_spec = 'TYPE' else: node.type_spec = stmt.__class__.__name__.upper() node.attrspec = attrspec node.selector = stmt.selector node.entity_decls = entity_decls node.kgen_use_tokgen = True #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) if len(out_entity_names) > 0: attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable', 'dimension']) entity_decls = get_decls(out_entity_names, stmt.entity_decls, prefix='kgenref_') attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) is_class_derived = check_class_derived(stmt) for entity_name, entity_decl in zip(entity_names, stmt.entity_decls): if node.kgen_parent.name + entity_name in self.kernel_extern_reads: continue if is_remove_state(entity_name, stmt): continue self.kernel_extern_reads.append(node.kgen_parent.name + entity_name) var = stmt.get_variable(entity_name) subrname = get_typedecl_readname(stmt, entity_name) if var.is_array(): if is_zero_array(var, stmt): continue if stmt.is_derived() or is_class_derived: self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, parent, var, stmt) self.kernel_created_subrs.append(subrname) else: # intrinsic type if var.is_explicit_shape_array(): self.create_read_intrinsic( self.kernel_externs_subrs[node.kgen_parent][0], entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_intrinsic( self.kernel_externs_subrs[node.kgen_parent][1], entity_name, stmt, var, prefix='kgenref_') else: # implicit array self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, parent, var, stmt) self.kernel_created_subrs.append(subrname) else: # scalar if stmt.is_derived() or is_class_derived: if var.is_allocatable() or var.is_pointer(): self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, parent, var, stmt) self.kernel_created_subrs.append(subrname) else: subrname = None for uname, req in stmt.unknowns.iteritems(): if (is_class_derived and uname.firstpartname() == stmt.selector[1] ) or uname.firstpartname() == stmt.name: #if uname.firstpartname()==stmt.name: if len(req.res_stmts) > 0: res = req.res_stmts[0] subrname = get_dtype_readname(res) break if subrname is None: print 'WARNING: Can not find Type resolver for %s' % stmt.name namedpart_append_comment(self.kernel_externs_subrs[node.kgen_parent][0], EXEC_PART, \ 'ERROR: "%s" is not resolved. Call statements to read "%s" is not created here.'%\ (stmt.name, stmt.name)) else: self.create_read_call( self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call( self.kernel_externs_subrs[ node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') else: # intrinsic type self.create_read_intrinsic( self.kernel_externs_subrs[node.kgen_parent][0], entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_intrinsic( self.kernel_externs_subrs[node.kgen_parent][1], entity_name, stmt, var, prefix='kgenref_')
def create_subr_read_typedecl_in_parentblock(self, node): stmt = node.kgen_stmt argintype = [] localintype = [] localouttype = [] for uname, req in KGGenType.get_state_in(stmt.geninfo): entity_name = uname.firstpartname() var = stmt.get_variable(entity_name) if var.is_parameter(): continue if is_remove_state(entity_name, stmt): continue if self.check_intent(entity_name, stmt): if (entity_name,DRIVER_READ_IN_ARGS) not in argintype: argintype.append((entity_name, DRIVER_READ_IN_ARGS)) if not entity_name in getinfo('kernel_driver_callsite_args'): getinfo('kernel_driver_callsite_args').append(entity_name) # add typedecl in driver attrs={'type_spec':stmt.__class__.__name__.upper(), 'selector':stmt.selector, 'entity_decls': [entity_name]} attrspec = [] if var.is_array(): attrspec.append('DIMENSION(%s)'%','.join(':'*var.rank)) if not var.is_pointer(): attrspec.append('ALLOCATABLE') # deallocate if var.is_pointer(): attrspec.append('POINTER') attrs['attrspec'] = attrspec namedpart_append_genknode(node.kgen_kernel_id, DRIVER_DECL_PART, stmt.__class__, attrs=attrs) if hasattr(stmt, 'unknowns'): for uname, req in stmt.unknowns.iteritems(): if len(req.res_stmts)>0: if req.res_stmts[-1].__class__==statements.Use: checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.name==req.res_stmts[-1].name \ and ( n.kgen_stmt.isonly and uname.firstpartname() in [ item.split('=>')[0].strip() for item in n.kgen_stmt.items]) if not namedpart_has_node(node.kgen_kernel_id, DRIVER_USE_PART, checks): item_name = uname.firstpartname() for new_name, old_name in req.res_stmts[-1].renames: if new_name==item_name: item_name = '%s => %s'%(new_name, old_name) break if not (req.res_stmts[-1].name, item_name) in self.driver_created_uses: attrs = {'name':req.res_stmts[-1].name, 'isonly': True, 'items':[item_name]} namedpart_append_genknode(node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append((req.res_stmts[-1].name, item_name)) if stmt.is_derived() and stmt.name==uname.firstpartname(): readname = get_dtype_readname(req.res_stmts[0]) if not (req.res_stmts[-1].name, readname) in self.driver_created_uses: attrs = {'name':req.res_stmts[-1].name, 'isonly': True, 'items':[readname]} namedpart_append_genknode(node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append((req.res_stmts[-1].name, readname)) else: if req.res_stmts[0].genkpair.kgen_parent!=node.kgen_parent: checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.name==get_topname(req.res_stmts[-1]) and \ ( n.kgen_stmt.isonly and uname.firstpartname() in [ item.split('=>')[0].strip() for item in n.kgen_stmt.items]) if not namedpart_has_node(node.kgen_kernel_id, DRIVER_USE_PART, checks): item_name = uname.firstpartname() for new_name, old_name in req.res_stmts[-1].renames: if new_name==item_name: item_name = '%s => %s'%(new_name, old_name) break if not (get_topname(req.res_stmts[-1]), item_name) in self.driver_created_uses: attrs = {'name':get_topname(req.res_stmts[-1]), 'isonly': True, 'items':[item_name]} namedpart_append_genknode(node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append((get_topname(req.res_stmts[-1]), item_name)) if stmt.is_derived() and stmt.name==uname.firstpartname(): readname = get_dtype_readname(req.res_stmts[-1]) if not (get_topname(req.res_stmts[-1]), readname) in self.driver_created_uses: attrs = {'name':get_topname(req.res_stmts[-1]), 'isonly': True, 'items':[readname]} namedpart_append_genknode(node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append((get_topname(req.res_stmts[-1]), readname)) elif (entity_name,KERNEL_PBLOCK_READ_IN_LOCALS) not in localintype and (entity_name,DRIVER_READ_IN_ARGS) not in argintype: localintype.append((uname.firstpartname(), KERNEL_PBLOCK_READ_IN_LOCALS)) for uname, req in KGGenType.get_state_out(stmt.geninfo): entity_name = uname.firstpartname() var = stmt.get_variable(entity_name) if var.is_parameter(): continue if is_remove_state(entity_name, stmt): continue if (entity_name,KERNEL_PBLOCK_READ_OUT_LOCALS) not in localouttype: localouttype.append((uname.firstpartname(), KERNEL_PBLOCK_READ_OUT_LOCALS)) if (entity_name,DRIVER_READ_IN_ARGS) in argintype: continue if (entity_name,KERNEL_PBLOCK_READ_IN_LOCALS) not in localintype: localintype.append((uname.firstpartname(), KERNEL_PBLOCK_READ_IN_LOCALS)) localvartypes = { 'localintype': localintype, 'localouttype': localouttype } def get_attrs(attrspec, allowed_attrs): attrspec = [] for attr in stmt.attrspec: if any( attr.startswith(allowed_attr) for allowed_attr in allowed_attrs): attrspec.append(attr) return attrspec def get_decls(names, decls, prefix=''): import re entity_decls = [] for decl in decls: ename = re.split('\(|\*|=', decl)[0].strip() if ename in names: entity_decls.append(prefix+decl) return entity_decls def get_enames(names, decls, prefix=''): import re entity_enames = [] for decl in decls: ename = re.split('\(|\*|=', decl)[0].strip() if ename in names: entity_enames.append(prefix+ename) return entity_enames if len(argintype)>0: attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable', 'dimension', 'target']) attrspec.append('INTENT(INOUT)') argin_names = [ argin_name for argin_name, pname in argintype] entity_decls = get_decls(argin_names, stmt.entity_decls) attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} if stmt.is_derived(): node.type_spec = 'TYPE' else: node.type_spec = stmt.__class__.__name__.upper() node.attrspec = attrspec node.selector = stmt.selector node.entity_decls = entity_decls node.kgen_use_tokgen = True #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) if len(localintype)==0 and len(argintype)==0 and len(localouttype)==0: node.kgen_forced_line = False elif len(localintype)>0: attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable', 'dimension', 'target']) localin_names = [ localin_name for localin_name, pname in localintype] entity_decls = get_decls(localin_names, stmt.entity_decls) attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} if stmt.is_derived(): node.type_spec = 'TYPE' else: node.type_spec = stmt.__class__.__name__.upper() node.attrspec = attrspec node.selector = stmt.selector node.entity_decls = entity_decls node.kgen_use_tokgen = True #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) local_allocate = False if len(localouttype)>0: for localout_name, partid in localouttype: attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable']) var = stmt.get_variable(localout_name) if var.is_array(): if var.is_explicit_shape_array(): attrspec.append('dimension(%s)'%','.join(var.shape)) else: attrspec.append('dimension(%s)'%','.join([':']*var.rank)) if 'allocatable' not in attrspec and 'pointer' not in attrspec: local_allocate = True attrspec.append('allocatable') #localout_names = [ localout_name for localout_name, pname in localouttype] #entity_decls = get_decls(localout_names, stmt.entity_decls, prefix='kgenref_') #entity_enames = get_enames(localout_names, stmt.entity_decls, prefix='kgenref_') attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': [ 'kgenref_%s'%localout_name ]} #'selector':stmt.selector, 'entity_decls': entity_decls} part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) # for kernel - local variables is_class_derived = check_class_derived(stmt) for vartypename, vartype in localvartypes.iteritems(): for entity_name, partid in vartype: if vartypename=='localouttype': ename_prefix = 'kgenref_' else: ename_prefix = '' var = stmt.get_variable(entity_name) subrname = get_typedecl_readname(stmt, entity_name) if var.is_array(): if is_zero_array(var, stmt): continue if stmt.is_derived() or is_class_derived: self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: # intrinsic type if var.is_explicit_shape_array(): if var.is_pointer(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: self.create_read_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var, ename_prefix=ename_prefix) else: # implicit array self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: # scalar if stmt.is_derived() or is_class_derived or var.is_pointer(): if var.is_allocatable() or var.is_pointer(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: subrname = None for uname, req in stmt.unknowns.iteritems(): if uname.firstpartname()==stmt.name and len(req.res_stmts)>0: res = req.res_stmts[0] subrname = get_dtype_readname(res) break if subrname is None: print 'WARNING: Can not find Type resolver for %s'%stmt.name namedpart_append_comment(node.kgen_kernel_id, partid, \ 'ERROR: "%s" is not resolved. Call statements to read "%s" is not created here.'%\ (stmt.name, stmt.name)) else: self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) else: # intrinsic type self.create_read_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var, ename_prefix=ename_prefix) # for kernel - argument variables for entity_name, partid in argintype: var = stmt.get_variable(entity_name) subrname = get_typedecl_readname(stmt, entity_name) if var.is_array(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var) if subrname not in self.driver_created_subrs: create_read_subr(subrname, entity_name, shared_objects['driver_object'], var, stmt, allocate=True) self.driver_created_subrs.append(subrname) else: # scalar if stmt.is_derived() or is_class_derived or var.is_pointer(): if var.is_allocatable() or var.is_pointer(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix) self.kernel_created_subrs.append(subrname) else: subrname = None for uname, req in stmt.unknowns.iteritems(): if uname.firstpartname()==stmt.name and len(req.res_stmts)>0: res = req.res_stmts[0] subrname = get_dtype_readname(res) break if subrname is None: print 'WARNING: Can not find Type resolver for %s'%stmt.name namedpart_append_comment(node.kgen_kernel_id, partid, \ 'ERROR: "%s" is not resolved. Call statements to read "%s" is not created here.'%\ (stmt.name, stmt.name)) else: self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var) else: # intrinsic type self.create_read_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var)
def create_subr_read_typedecl_in_parentblock(self, node): stmt = node.kgen_stmt argintype = [] localintype = [] localouttype = [] for uname, req in KGGenType.get_state_in(stmt.geninfo): entity_name = uname.firstpartname() var = stmt.get_variable(entity_name) if var.is_parameter(): continue if is_remove_state(entity_name, stmt): continue if self.check_intent(entity_name, stmt): if (entity_name, DRIVER_READ_IN_ARGS) not in argintype: argintype.append((entity_name, DRIVER_READ_IN_ARGS)) if not entity_name in getinfo( 'kernel_driver_callsite_args'): getinfo('kernel_driver_callsite_args').append( entity_name) # add typedecl in driver attrs = { 'type_spec': stmt.__class__.__name__.upper(), 'selector': stmt.selector, 'entity_decls': [entity_name] } attrspec = [] if var.is_array(): attrspec.append('DIMENSION(%s)' % ','.join(':' * var.rank)) if not var.is_pointer(): attrspec.append('ALLOCATABLE') # deallocate if var.is_pointer(): attrspec.append('POINTER') attrs['attrspec'] = attrspec namedpart_append_genknode(node.kgen_kernel_id, DRIVER_DECL_PART, stmt.__class__, attrs=attrs) if hasattr(stmt, 'unknowns'): for uname, req in stmt.unknowns.iteritems(): if len(req.res_stmts) > 0: if req.res_stmts[ -1].__class__ == statements.Use: checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.name==req.res_stmts[-1].name \ and ( n.kgen_stmt.isonly and uname.firstpartname() in [ item.split('=>')[0].strip() for item in n.kgen_stmt.items]) if not namedpart_has_node( node.kgen_kernel_id, DRIVER_USE_PART, checks): item_name = uname.firstpartname() for new_name, old_name in req.res_stmts[ -1].renames: if new_name == item_name: item_name = '%s => %s' % ( new_name, old_name) break if not (req.res_stmts[-1].name, item_name ) in self.driver_created_uses: attrs = { 'name': req.res_stmts[-1].name, 'isonly': True, 'items': [item_name] } namedpart_append_genknode( node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append( (req.res_stmts[-1].name, item_name)) if stmt.is_derived( ) and stmt.name == uname.firstpartname( ): readname = get_dtype_readname( req.res_stmts[0]) if not ( req.res_stmts[-1].name, readname ) in self.driver_created_uses: attrs = { 'name': req.res_stmts[-1].name, 'isonly': True, 'items': [readname] } namedpart_append_genknode( node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append( (req.res_stmts[-1].name, readname)) else: if req.res_stmts[ 0].genkpair.kgen_parent != node.kgen_parent: checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.name==get_topname(req.res_stmts[-1]) and \ ( n.kgen_stmt.isonly and uname.firstpartname() in [ item.split('=>')[0].strip() for item in n.kgen_stmt.items]) if not namedpart_has_node( node.kgen_kernel_id, DRIVER_USE_PART, checks): item_name = uname.firstpartname() for new_name, old_name in req.res_stmts[ -1].renames: if new_name == item_name: item_name = '%s => %s' % ( new_name, old_name) break if not ( get_topname( req.res_stmts[-1]), item_name ) in self.driver_created_uses: attrs = { 'name': get_topname( req.res_stmts[-1]), 'isonly': True, 'items': [item_name] } namedpart_append_genknode( node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append( (get_topname( req.res_stmts[-1]), item_name)) if stmt.is_derived( ) and stmt.name == uname.firstpartname( ): readname = get_dtype_readname( req.res_stmts[-1]) if not ( get_topname( req.res_stmts[-1]), readname ) in self.driver_created_uses: attrs = { 'name': get_topname( req.res_stmts[-1]), 'isonly': True, 'items': [readname] } namedpart_append_genknode( node.kgen_kernel_id, DRIVER_USE_PART, statements.Use, attrs=attrs) self.driver_created_uses.append( (get_topname( req.res_stmts[-1]), readname)) elif (entity_name, KERNEL_PBLOCK_READ_IN_LOCALS) not in localintype and ( entity_name, DRIVER_READ_IN_ARGS) not in argintype: localintype.append( (uname.firstpartname(), KERNEL_PBLOCK_READ_IN_LOCALS)) for uname, req in KGGenType.get_state_out(stmt.geninfo): entity_name = uname.firstpartname() var = stmt.get_variable(entity_name) if var.is_parameter(): continue if is_remove_state(entity_name, stmt): continue if (entity_name, KERNEL_PBLOCK_READ_OUT_LOCALS) not in localouttype: localouttype.append( (uname.firstpartname(), KERNEL_PBLOCK_READ_OUT_LOCALS)) if (entity_name, DRIVER_READ_IN_ARGS) in argintype: continue if (entity_name, KERNEL_PBLOCK_READ_IN_LOCALS) not in localintype: localintype.append( (uname.firstpartname(), KERNEL_PBLOCK_READ_IN_LOCALS)) localvartypes = { 'localintype': localintype, 'localouttype': localouttype } def get_attrs(attrspec, allowed_attrs): attrspec = [] for attr in stmt.attrspec: if any( attr.startswith(allowed_attr) for allowed_attr in allowed_attrs): attrspec.append(attr) return attrspec def get_decls(names, decls, prefix=''): import re entity_decls = [] for decl in decls: ename = re.split('\(|\*|=', decl)[0].strip() if ename in names: entity_decls.append(prefix + decl) return entity_decls def get_enames(names, decls, prefix=''): import re entity_enames = [] for decl in decls: ename = re.split('\(|\*|=', decl)[0].strip() if ename in names: entity_enames.append(prefix + ename) return entity_enames if len(argintype) > 0: attrspec = get_attrs( stmt.attrspec, ['pointer', 'allocatable', 'dimension', 'target']) attrspec.append('INTENT(INOUT)') argin_names = [argin_name for argin_name, pname in argintype] entity_decls = get_decls(argin_names, stmt.entity_decls) attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} if stmt.is_derived(): node.type_spec = 'TYPE' else: node.type_spec = stmt.__class__.__name__.upper() node.attrspec = attrspec node.selector = stmt.selector node.entity_decls = entity_decls node.kgen_use_tokgen = True #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) if len(localintype) == 0 and len(argintype) == 0 and len( localouttype) == 0: node.kgen_forced_line = False elif len(localintype) > 0: attrspec = get_attrs( stmt.attrspec, ['pointer', 'allocatable', 'dimension', 'target']) localin_names = [ localin_name for localin_name, pname in localintype ] entity_decls = get_decls(localin_names, stmt.entity_decls) attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} if stmt.is_derived(): node.type_spec = 'TYPE' else: node.type_spec = stmt.__class__.__name__.upper() node.attrspec = attrspec node.selector = stmt.selector node.entity_decls = entity_decls node.kgen_use_tokgen = True #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) local_allocate = False if len(localouttype) > 0: for localout_name, partid in localouttype: attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable']) var = stmt.get_variable(localout_name) if var.is_array(): if var.is_explicit_shape_array(): attrspec.append('dimension(%s)' % ','.join([ ':'.join(slice) for slice in var.get_array_spec() ])) else: attrspec.append('dimension(%s)' % ','.join([':'] * var.rank)) if 'allocatable' not in attrspec and 'pointer' not in attrspec: local_allocate = True attrspec.append('allocatable') #localout_names = [ localout_name for localout_name, pname in localouttype] #entity_decls = get_decls(localout_names, stmt.entity_decls, prefix='kgenref_') #entity_enames = get_enames(localout_names, stmt.entity_decls, prefix='kgenref_') attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': [ 'kgenref_%s'%localout_name ]} #'selector':stmt.selector, 'entity_decls': entity_decls} part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) # for kernel - local variables is_class_derived = check_class_derived(stmt) for vartypename, vartype in localvartypes.iteritems(): for entity_name, partid in vartype: if vartypename == 'localouttype': ename_prefix = 'kgenref_' else: ename_prefix = '' var = stmt.get_variable(entity_name) subrname = get_typedecl_readname(stmt, entity_name) if var.is_array(): if is_zero_array(var, stmt): continue if stmt.is_derived() or is_class_derived: self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: # intrinsic type if var.is_explicit_shape_array(): if var.is_pointer(): self.create_read_call( node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: self.create_read_intrinsic( node.kgen_kernel_id, partid, entity_name, stmt, var, ename_prefix=ename_prefix) else: # implicit array self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: # scalar if stmt.is_derived() or is_class_derived or var.is_pointer( ): if var.is_allocatable() or var.is_pointer(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix, allocate=local_allocate) self.kernel_created_subrs.append(subrname) else: subrname = None for uname, req in stmt.unknowns.iteritems(): if uname.firstpartname() == stmt.name and len( req.res_stmts) > 0: res = req.res_stmts[0] subrname = get_dtype_readname(res) break if subrname is None: print 'WARNING: Can not find Type resolver for %s' % stmt.name namedpart_append_comment(node.kgen_kernel_id, partid, \ 'ERROR: "%s" is not resolved. Call statements to read "%s" is not created here.'%\ (stmt.name, stmt.name)) else: self.create_read_call( node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) else: # intrinsic type self.create_read_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var, ename_prefix=ename_prefix) # for kernel - argument variables for entity_name, partid in argintype: var = stmt.get_variable(entity_name) subrname = get_typedecl_readname(stmt, entity_name) if var.is_array(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var) if subrname not in self.driver_created_subrs: create_read_subr(subrname, entity_name, shared_objects['driver_object'], var, stmt, allocate=True) self.driver_created_subrs.append(subrname) else: # scalar if stmt.is_derived() or is_class_derived or var.is_pointer(): if var.is_allocatable() or var.is_pointer(): self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var, ename_prefix=ename_prefix) if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, node.kgen_parent, var, stmt, ename_prefix=ename_prefix) self.kernel_created_subrs.append(subrname) else: subrname = None for uname, req in stmt.unknowns.iteritems(): if uname.firstpartname() == stmt.name and len( req.res_stmts) > 0: res = req.res_stmts[0] subrname = get_dtype_readname(res) break if subrname is None: print 'WARNING: Can not find Type resolver for %s' % stmt.name namedpart_append_comment(node.kgen_kernel_id, partid, \ 'ERROR: "%s" is not resolved. Call statements to read "%s" is not created here.'%\ (stmt.name, stmt.name)) else: self.create_read_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var) else: # intrinsic type self.create_read_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var)
def create_subr_read_typedecl_in_module(self, node): parent = node.kgen_parent stmt = node.kgen_stmt raw_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state(stmt.geninfo)]) entity_names = [ e for e in raw_entity_names if not stmt.get_variable(e).is_parameter() ] raw_out_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state_out(stmt.geninfo)]) out_entity_names = [ e for e in raw_out_entity_names if not stmt.get_variable(e).is_parameter() ] def get_attrs(attrspec, allowed_attrs): attrspec = [] for attr in stmt.attrspec: if any( attr.startswith(allowed_attr) for allowed_attr in allowed_attrs): attrspec.append(attr) return attrspec def get_decls(names, decls, prefix=''): import re entity_decls = [] for decl in decls: ename = re.split('\(|\*|=', decl)[0].strip() if ename in names: entity_decls.append(prefix+decl) return entity_decls if len(entity_names)==0: node.kgen_forced_line = False elif len(entity_names)!=len(stmt.entity_decls): attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable', 'dimension', 'public', 'target']) entity_decls = get_decls(entity_names, stmt.entity_decls) attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} if stmt.is_derived(): node.type_spec = 'TYPE' else: node.type_spec = stmt.__class__.__name__.upper() node.attrspec = attrspec node.selector = stmt.selector node.entity_decls = entity_decls node.kgen_use_tokgen = True #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) #part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) if len(out_entity_names)>0: attrspec = get_attrs(stmt.attrspec, ['pointer', 'allocatable', 'dimension']) entity_decls = get_decls(out_entity_names, stmt.entity_decls, prefix='kgenref_') attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, \ 'selector':stmt.selector, 'entity_decls': entity_decls} part_append_genknode(node.kgen_parent, DECL_PART, stmt.__class__, attrs=attrs) is_class_derived = check_class_derived(stmt) for entity_name, entity_decl in zip(entity_names, stmt.entity_decls): if node.kgen_parent.name+entity_name in self.kernel_extern_reads: continue if is_remove_state(entity_name, stmt): continue self.kernel_extern_reads.append(node.kgen_parent.name+entity_name) var = stmt.get_variable(entity_name) subrname = get_typedecl_readname(stmt, entity_name) if var.is_array(): if is_zero_array(var, stmt): continue if stmt.is_derived() or is_class_derived: self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, parent, var, stmt) self.kernel_created_subrs.append(subrname) else: # intrinsic type if var.is_explicit_shape_array(): self.create_read_intrinsic(self.kernel_externs_subrs[node.kgen_parent][0], entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_intrinsic(self.kernel_externs_subrs[node.kgen_parent][1], entity_name, stmt, var, prefix='kgenref_') else: # implicit array self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, parent, var, stmt) self.kernel_created_subrs.append(subrname) else: # scalar if stmt.is_derived() or is_class_derived: if var.is_allocatable() or var.is_pointer(): self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') if subrname not in self.kernel_created_subrs: create_read_subr(subrname, entity_name, parent, var, stmt) self.kernel_created_subrs.append(subrname) else: subrname = None for uname, req in stmt.unknowns.iteritems(): if ( is_class_derived and uname.firstpartname()==stmt.selector[1]) or uname.firstpartname()==stmt.name: #if uname.firstpartname()==stmt.name: if len(req.res_stmts)>0: res = req.res_stmts[0] subrname = get_dtype_readname(res) break if subrname is None: print 'WARNING: Can not find Type resolver for %s'%stmt.name namedpart_append_comment(self.kernel_externs_subrs[node.kgen_parent][0], EXEC_PART, \ 'ERROR: "%s" is not resolved. Call statements to read "%s" is not created here.'%\ (stmt.name, stmt.name)) else: self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_call(self.kernel_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var, prefix='kgenref_') else: # intrinsic type self.create_read_intrinsic(self.kernel_externs_subrs[node.kgen_parent][0], entity_name, stmt, var) if entity_name in out_entity_names: self.create_read_intrinsic(self.kernel_externs_subrs[node.kgen_parent][1], entity_name, stmt, var, prefix='kgenref_')