def test_lfric_stub_args():
'''Check that correct stub code is produced when there are multiple
stencils.
'''
ast = get_ast("dynamo0.3", "testkern_stencil_multi_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "field_1_w1: READ+WRITE" in var_info
assert "field_2_stencil_dofmap: READ" in var_info
assert "field_2_stencil_size: READ" in var_info
assert "field_2_w2: READ" in var_info
assert "field_3_direction: READ" in var_info
assert "field_3_stencil_dofmap: READ" in var_info
assert "field_3_stencil_size: READ" in var_info
assert "field_3_w2: READ" in var_info
assert "field_4_stencil_dofmap: READ" in var_info
assert "field_4_stencil_size: READ" in var_info
assert "field_4_w3: READ" in var_info
assert "map_w1: READ" in var_info
assert "map_w2: READ" in var_info
assert "map_w3: READ" in var_info
assert "ndf_w1: READ" in var_info
assert "ndf_w2: READ" in var_info
assert "ndf_w3: READ" in var_info
assert "nlayers: READ" in var_info
assert "undf_w1: READ" in var_info
assert "undf_w2: READ" in var_info
assert "undf_w3: READ" in var_info
def test_lfric_stub_cma_operators():
'''Check variable usage detection cma operators.
mesh_ncell2d, cma_operator
'''
from psyclone.dynamo0p3 import DynKernMetadata, DynKern
from psyclone.domain.lfric import KernStubArgList
ast = get_ast("dynamo0.3", "columnwise_op_mul_2scalars_kernel_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
for num in ["1", "3", "5"]:
assert "ncell_2d: READ" in var_info
assert "cma_op_"+num+": READ" in var_info
assert "cma_op_"+num+"_nrow: READ" in var_info
assert "cma_op_"+num+"_ncol: READ" in var_info
assert "cma_op_"+num+"_bandwidth: READ" in var_info
assert "cma_op_"+num+"_alpha: READ" in var_info
assert "cma_op_"+num+"_beta: READ" in var_info
assert "cma_op_"+num+"_gamma_m: READ" in var_info
assert "cma_op_"+num+"_gamma_p: READ" in var_info
def test_lfric_stub_boundary_dofs():
'''Check variable usage detection for boundary dofs.
'''
ast = get_ast("dynamo0.3", "enforce_bc_kernel_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
assert "boundary_dofs_field_1: READ" in str(var_accesses)
def test_lfric_stub_indirection_dofmap():
'''Check variable usage detection in indirection dofmap.
'''
ast = get_ast("dynamo0.3", "columnwise_op_app_kernel_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "cma_indirection_map_aspc1_field_1: READ" in var_info
assert "cma_indirection_map_aspc2_field_2: READ" in var_info
def test_lfric_stub_banded_dofmap():
'''Check variable usage detection for banded dofmaps.
'''
ast = get_ast("dynamo0.3", "columnwise_op_asm_kernel_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "cbanded_map_adspc1_op_1: READ" in var_info
assert "cbanded_map_adspc2_op_1: READ" in var_info
def test_lfric_stub_boundary_dofmap():
'''Check variable usage detection in boundary_dofs array fix
for operators.
'''
from psyclone.dynamo0p3 import DynKernMetadata, DynKern
from psyclone.domain.lfric import KernStubArgList
ast = get_ast("dynamo0.3", "enforce_operator_bc_kernel_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
assert "boundary_dofs_op_1: READ" in str(var_accesses)
def test_kernstubarglist_arglist_error():
'''Check that we raise an exception if we call the arglist method in
kernstubarglist without first calling the generate method'''
ast = get_ast(TEST_API, "testkern_one_int_scalar_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
# Now call KernStubArgList to raise an exception
create_arg_list = KernStubArgList(kernel)
with pytest.raises(InternalError) as excinfo:
_ = create_arg_list.arglist
assert ("The argument list in KernStubArgList is "
"empty. Has the generate() method been "
"called?") in str(excinfo.value)
def test_lfric_stub_args3():
'''Check variable usage detection for cell position, operator
'''
ast = get_ast("dynamo0.3", "testkern_any_discontinuous_space_op_1_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "cell: READ" in var_info
assert "op_3: READ" in var_info
assert "op_3_ncell_3d: READ" in var_info
assert "op_4: READ" in var_info
assert "op_4_ncell_3d: READ" in var_info
def test_kernel_stub_invalid_scalar_argument():
''' Check that we raise an exception if an unexpected datatype is found
when using the KernStubArgList scalar method. '''
ast = get_ast(TEST_API, "testkern_one_int_scalar_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
# Sabotage the scalar argument to make it have an invalid type.
arg = kernel.arguments.args[1]
arg._argument_type = "invalid"
# Now call KernStubArgList to raise an exception
create_arg_list = KernStubArgList(kernel)
with pytest.raises(InternalError) as excinfo:
create_arg_list.scalar(arg)
const = LFRicConstants()
assert ("Expected argument type to be one of {0} but got "
"'invalid'".format(const.VALID_SCALAR_NAMES) in str(excinfo.value))
def test_lfric_stub_basis():
'''Check variable usage detection of basis, diff-basis.
'''
ast = get_ast("dynamo0.3", "testkern_qr_eval_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "basis_w1_on_w1: READ" in var_info
assert "diff_basis_w2_qr_face: READ" in var_info
assert "diff_basis_w2_on_w1: READ" in var_info
assert "basis_w3_on_w1: READ" in var_info
assert "diff_basis_w3_qr_face: READ" in var_info
assert "diff_basis_w3_on_w1: READ" in var_info
def test_lfric_stub_args2():
'''Check variable usage detection for scalars, basis_name, quad rule
and mesh properties.
'''
ast = get_ast("dynamo0.3", "testkern_mesh_prop_face_qr_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "rscalar_1: READ" in var_info
assert "basis_w1_qr_face: READ" in var_info
assert "nfaces_qr_face: READ" in var_info
assert "np_xyz_qr_face: READ" in var_info
assert "weights_xyz_qr_face: READ" in var_info
def test_refelem_stub_arglist_err():
''' Check that the KernStubArgList.ref_element_properties method raises
the expected error if it encounters an unsupported property. '''
# Create the Kernel object
ast = get_ast(TEST_API, "testkern_ref_elem_all_faces_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
# Break the list of ref-element properties required by the Kernel
kernel.reference_element.properties.append("Wrong property")
with pytest.raises(InternalError) as err:
KernStubArgList(kernel).generate()
assert "('Wrong property') " in str(err.value)
assert (
"Supported properties are: ['Property.NORMALS_TO_HORIZONTAL_FACES', "
"'Property.NORMALS_TO_VERTICAL_FACES', 'Property.NORMALS_TO_FACES', "
"'Property.OUTWARD_NORMALS_TO_HORIZONTAL_FACES', "
"'Property.OUTWARD_NORMALS_TO_VERTICAL_FACES', "
"'Property.OUTWARD_NORMALS_TO_FACES']" in str(err.value))
def test_kernstubarglist_eval_shape_error():
''' Check that we raise the expected exception if we call the basis() or
diff_basis() methods and one of the kernel's evaluator shapes is
invalid. '''
ast = get_ast(TEST_API, "testkern_qr_faces_mod.F90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
create_arg_list = KernStubArgList(kernel)
# Break the list of qr rules
kernel.eval_shapes.insert(0, "broken")
with pytest.raises(InternalError) as err:
create_arg_list.basis(None)
assert ("Unrecognised evaluator shape ('broken'). Expected one of: "
"['gh_quadrature_xyoz'" in str(err.value))
with pytest.raises(InternalError) as err:
create_arg_list.diff_basis(None)
assert ("Unrecognised evaluator shape ('broken'). Expected one of: "
"['gh_quadrature_xyoz'" in str(err.value))
def test_kernel_stub_ind_dofmap_errors():
'''Check that we raise the expected exceptions if the wrong arguments
are supplied to KernelStubArgList.indirection_dofmap() '''
ast = get_ast(TEST_API, "testkern_one_int_scalar_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
# Now call KernStubArgList to raise an exception
create_arg_list = KernStubArgList(kernel)
# First call it without an argument object
with pytest.raises(InternalError) as excinfo:
create_arg_list.indirection_dofmap("w3")
assert "No CMA operator supplied" in str(excinfo.value)
# Second, call it with an argument object but one that is not
# an operator
with pytest.raises(InternalError) as excinfo:
create_arg_list.indirection_dofmap("w3", kernel.arguments.args[1])
assert ("A CMA operator (gh_columnwise_operator) must be supplied but "
"got") in str(excinfo.value)
def test_lfric_stub_field_vector():
'''Check variable usage detection field vectors.
'''
ast = get_ast("dynamo0.3", "testkern_stencil_vector_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "field_1_w0_v1: READ" in var_info
assert "field_1_w0_v2: READ" in var_info
assert "field_1_w0_v3: READ" in var_info
assert "field_2_w3_v1: READ" in var_info
assert "field_2_w3_v2: READ" in var_info
assert "field_2_w3_v3: READ" in var_info
assert "field_2_w3_v4: READ" in var_info
def test_lfric_stub_args3():
'''Check variable usage detection for cell position, operator
'''
from psyclone.dynamo0p3 import DynKernMetadata, DynKern
from psyclone.domain.lfric import KernStubArgList
ast = get_ast("dynamo0.3", "dummy_orientation_mod.f90")
metadata = DynKernMetadata(ast)
kernel = DynKern()
kernel.load_meta(metadata)
var_accesses = VariablesAccessInfo()
create_arg_list = KernStubArgList(kernel)
create_arg_list.generate(var_accesses=var_accesses)
var_info = str(var_accesses)
assert "cell: READ" in var_info
assert "op_2_ncell_3d: READ" in var_info
assert "op_2: READ" in var_info
assert "op_4_ncell_3d: READ" in var_info
assert "op_4: READ" in var_info
assert "orientation_w0: READ" in var_info
assert "orientation_w1: READ" in var_info
assert "orientation_w2: READ" in var_info
assert "orientation_w3: READ" in var_info
