brintos

brintos / llvm-project-archived public Read only

0
0
Text · 8.0 KiB · 47c3468 Raw
313 lines · plain
1/* Copyright 2016-2017 Tobias Grosser2 *3 * Use of this software is governed by the MIT license4 *5 * Written by Tobias Grosser, Weststrasse 47, CH-8003, Zurich6 */7 8#include <vector>9#include <string>10#include <limits.h>11#include <stdio.h>12#include <stdlib.h>13 14#include <isl/options.h>15#include <isl/cpp-checked.h>16 17namespace isl { using namespace checked; }18 19static void assert_impl(bool condition, const char *file, int line,20	const char *message)21{22	if (condition)23		return;24 25	fprintf(stderr, "Assertion failed in %s:%d %s\n", file, line, message);26	exit(EXIT_FAILURE);27}28 29static void assert_impl(isl::boolean condition, const char *file, int line,30	const char *message)31{32	assert_impl(bool(condition), file, line, message);33}34 35/* Return the value encapsulated by "s".36 */37static int size_val(isl::size s)38{39	return s.is_error() ? -1 : unsigned(s);40}41 42#undef assert43#define assert(exp) assert_impl(exp, __FILE__, __LINE__, #exp)44#define IS_TRUE(b)	(b).is_true()45#define SIZE_VAL(s)	size_val(s)46 47#include "isl_test_cpp-generic.cc"48 49/* Test that isl_bool values are returned correctly.50 *51 * We check in detail the following parts of the isl::boolean class:52 *  - The is_true, is_false, and is_error functions return true in case they53 *    are called on a true, false, or error instance of isl::boolean,54 *    respectively55 *  - Explicit conversion to 'bool'56 *  - Implicit conversion to 'bool'57 *  - The complement operator58 *  - Explicit construction from 'true' and 'false'59 *  - Explicit construction form isl_bool60 */61void test_return_bool(isl::ctx ctx)62{63	isl::set empty(ctx, "{ : false }");64	isl::set univ(ctx, "{ : }");65	isl::set null;66 67	isl::boolean b_true = empty.is_empty();68	isl::boolean b_false = univ.is_empty();69	isl::boolean b_error = null.is_empty();70 71	assert(b_true.is_true());72	assert(!b_true.is_false());73	assert(!b_true.is_error());74 75	assert(!b_false.is_true());76	assert(b_false.is_false());77	assert(!b_false.is_error());78 79	assert(!b_error.is_true());80	assert(!b_error.is_false());81	assert(b_error.is_error());82 83	assert(bool(b_true) == true);84	assert(bool(b_false) == false);85 86	assert(b_true);87 88	assert((!b_false).is_true());89	assert((!b_true).is_false());90	assert((!b_error).is_error());91 92	assert(isl::boolean(true).is_true());93	assert(!isl::boolean(true).is_false());94	assert(!isl::boolean(true).is_error());95 96	assert(isl::boolean(false).is_false());97	assert(!isl::boolean(false).is_true());98	assert(!isl::boolean(false).is_error());99 100	assert(isl::manage(isl_bool_true).is_true());101	assert(!isl::manage(isl_bool_true).is_false());102	assert(!isl::manage(isl_bool_true).is_error());103 104	assert(isl::manage(isl_bool_false).is_false());105	assert(!isl::manage(isl_bool_false).is_true());106	assert(!isl::manage(isl_bool_false).is_error());107 108	assert(isl::manage(isl_bool_error).is_error());109	assert(!isl::manage(isl_bool_error).is_true());110	assert(!isl::manage(isl_bool_error).is_false());111}112 113/* Test that return values are handled correctly.114 *115 * Test that isl C++ objects, integers, boolean values, and strings are116 * returned correctly.117 */118void test_return(isl::ctx ctx)119{120	test_return_obj(ctx);121	test_return_int(ctx);122	test_return_bool(ctx);123	test_return_string(ctx);124}125 126/* Test that foreach functions are modeled correctly.127 *128 * Verify that lambdas are correctly called as callback of a 'foreach'129 * function and that variables captured by the lambda work correctly. Also130 * check that the foreach function takes account of the return value of the131 * lambda and aborts in case isl::stat::error is returned and then returns132 * isl::stat::error itself.133 */134void test_foreach(isl::ctx ctx)135{136	isl::set s(ctx, "{ [0]; [1]; [2] }");137 138	std::vector<isl::basic_set> basic_sets;139 140	auto add_to_vector = [&] (isl::basic_set bs) {141		basic_sets.push_back(bs);142		return isl::stat::ok();143	};144 145	isl::stat ret1 = s.foreach_basic_set(add_to_vector);146 147	assert(ret1.is_ok());148	assert(basic_sets.size() == 3);149	assert(isl::set(basic_sets[0]).is_subset(s).is_true());150	assert(isl::set(basic_sets[1]).is_subset(s).is_true());151	assert(isl::set(basic_sets[2]).is_subset(s).is_true());152	assert(!basic_sets[0].is_equal(basic_sets[1]).is_true());153 154	auto fail = [&] (isl::basic_set bs) {155		return isl::stat::error();156	};157 158	isl::stat ret2 = s.foreach_basic_set(fail);159 160	assert(ret2.is_error());161}162 163/* Test the functionality of "every" functions.164 *165 * In particular, test the generic functionality and166 * test that error conditions are properly propagated.167 */168static void test_every(isl::ctx ctx)169{170	isl::union_set us(ctx, "{ A[i]; B[j] }");171 172	test_every_generic(ctx);173 174	auto fail = [] (isl::set s){175		return isl::boolean::error();176	};177	assert(us.every_set(fail).is_error());178}179 180/* Test basic schedule tree functionality.181 *182 * In particular, create a simple schedule tree and183 * - perform some generic tests184 * - test map_descendant_bottom_up in the failing case185 * - test foreach_descendant_top_down186 * - test every_descendant187 */188static void test_schedule_tree(isl::ctx ctx)189{190	auto root = test_schedule_tree_generic(ctx);191 192	auto fail_map = [](isl::schedule_node node) {193		return isl::schedule_node();194	};195	assert(root.map_descendant_bottom_up(fail_map).is_null());196 197	int count = 0;198	auto inc_count = [&count](isl::schedule_node node) {199		count++;200		return isl::boolean(true);201	};202	assert(root.foreach_descendant_top_down(inc_count).is_ok());203	assert(count == 8);204 205	count = 0;206	auto inc_count_once = [&count](isl::schedule_node node) {207		count++;208		return isl::boolean(false);209	};210	assert(root.foreach_descendant_top_down(inc_count_once).is_ok());211	assert(count == 1);212 213	auto is_not_domain = [](isl::schedule_node node) {214		return !node.isa<isl::schedule_node_domain>();215	};216	assert(root.child(0).every_descendant(is_not_domain).is_true());217	assert(root.every_descendant(is_not_domain).is_false());218 219	auto fail = [](isl::schedule_node node) {220		return isl::boolean();221	};222	assert(root.every_descendant(fail).is_error());223 224	auto domain = root.as<isl::schedule_node_domain>().domain();225	auto filters = isl::union_set(ctx, "{}");226	auto collect_filters = [&filters](isl::schedule_node node) {227		if (node.isa<isl::schedule_node_filter>().is_true()) {228			auto filter = node.as<isl::schedule_node_filter>();229			filters = filters.unite(filter.filter());230		}231		return isl::boolean(true);232	};233	assert(!root.every_descendant(collect_filters).is_error());234	assert(domain.is_equal(filters).is_true());235}236 237/* Test basic AST generation from a schedule tree.238 *239 * In particular, create a simple schedule tree and240 * - perform some generic tests241 * - test at_each_domain in the failing case242 */243static void test_ast_build(isl::ctx ctx)244{245	auto schedule = test_ast_build_generic(ctx);246 247	bool do_fail = true;248	int count_ast_fail = 0;249	auto fail_inc_count_ast =250	    [&count_ast_fail, &do_fail](isl::ast_node node,251					isl::ast_build build) {252		count_ast_fail++;253		return do_fail ? isl::ast_node() : node;254	};255	auto build = isl::ast_build(ctx);256	build = build.set_at_each_domain(fail_inc_count_ast);257	auto ast = build.node_from(schedule);258	assert(ast.is_null());259	assert(count_ast_fail > 0);260	auto build_copy = build;261	int count_ast = 0;262	auto inc_count_ast =263	    [&count_ast](isl::ast_node node, isl::ast_build build) {264		count_ast++;265		return node;266	};267	build_copy = build_copy.set_at_each_domain(inc_count_ast);268	ast = build_copy.node_from(schedule);269	assert(!ast.is_null());270	assert(count_ast == 2);271	count_ast_fail = 0;272	do_fail = false;273	ast = build.node_from(schedule);274	assert(!ast.is_null());275	assert(count_ast_fail == 2);276}277 278/* Test the isl checked C++ interface279 *280 * This includes:281 *  - The isl C <-> C++ pointer interface282 *  - Object construction283 *  - Different parameter types284 *  - Different return types285 *  - Foreach functions286 *  - Every functions287 *  - Spaces288 *  - Schedule trees289 *  - AST generation290 *  - AST expression generation291 */292int main()293{294	isl_ctx *ctx = isl_ctx_alloc();295 296	isl_options_set_on_error(ctx, ISL_ON_ERROR_ABORT);297 298	test_pointer(ctx);299	test_constructors(ctx);300	test_parameters(ctx);301	test_return(ctx);302	test_foreach(ctx);303	test_every(ctx);304	test_space(ctx);305	test_schedule_tree(ctx);306	test_ast_build(ctx);307	test_ast_build_expr(ctx);308 309	isl_ctx_free(ctx);310 311	return EXIT_SUCCESS;312}313