loki.transformations.tests.test_pool_allocator

Functions

check_c_sizeof_import(routine)

check_real64_import(routine)

check_stack_created_in_driver(driver, ...[, ...])

fixture_block_dim()

fixture_block_dim_alt()

fixture_horizontal()

test_pool_allocator_args_vs_kwargs(tmp_path, ...)

test_pool_allocator_more_call_checks(...)

test_pool_allocator_temporaries(tmp_path, ...)

test_pool_allocator_temporaries_kernel_nested(...)

test_pool_allocator_temporaries_kernel_sequence(...)

test_pool_allocator_unused_temporaries(...)

fixture_block_dim()
fixture_horizontal()
fixture_block_dim_alt()
check_c_sizeof_import(routine)
check_real64_import(routine)
check_stack_created_in_driver(driver, stack_size, first_kernel_call, num_block_loops, check_bounds=True, cray_ptr_loc_rhs=False)
test_pool_allocator_temporaries(tmp_path, frontend, generate_driver_stack, block_dim, check_bounds, nclv_param, cray_ptr_loc_rhs, horizontal)
test_pool_allocator_unused_temporaries(tmp_path, frontend, horizontal, block_dim)
test_pool_allocator_temporaries_kernel_sequence(tmp_path, frontend, block_dim, directive, stack_insert_pragma, cray_ptr_loc_rhs, horizontal)
test_pool_allocator_temporaries_kernel_nested(tmp_path, frontend, block_dim, directive, cray_ptr_loc_rhs)
test_pool_allocator_more_call_checks(tmp_path, frontend, block_dim, caplog, cray_ptr_loc_rhs, horizontal)
test_pool_allocator_args_vs_kwargs(tmp_path, frontend, block_dim_alt, cray_ptr_loc_rhs, horizontal)