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_')
