3 #ifndef LALA_CORE_VSTORE_HPP
4 #define LALA_CORE_VSTORE_HPP
14 static constexpr
bool atoms =
false;
18 static constexpr
bool atoms =
true;
37 template<
class U,
class Allocator>
45 template <
class Alloc>
53 template <
class VarDom>
57 template <
class Alloc>
58 using tell_type = battery::vector<var_dom<Alloc>, Alloc>;
60 template <
class Alloc>
63 template <
class Alloc = allocator_type>
67 constexpr
static const bool sequential = universe_type::sequential;
71 constexpr
static const bool preserve_join = universe_type::preserve_join;
72 constexpr
static const bool preserve_meet = universe_type::preserve_meet;
75 constexpr
static const char*
name =
"VStore";
77 template<
class U2,
class Alloc2>
81 using store_type = battery::vector<universe_type, allocator_type>;
82 using memory_type =
typename universe_type::memory_type;
90 : atype(other.atype), data(other.data), is_at_bot(other.is_at_bot)
95 : atype(atype), data(alloc), is_at_bot(false)
99 : atype(atype), data(size, alloc), is_at_bot(false)
104 : atype(other.atype), data(other.data), is_at_bot(other.is_at_bot)
107 template<
class R,
class Alloc2>
109 : atype(other.atype), data(other.data, alloc), is_at_bot(other.is_at_bot)
114 template<
class R,
class Alloc2,
class... Allocators>
119 atype(other.atype), data(std::move(other.data)), is_at_bot(other.is_at_bot) {}
122 return data.get_allocator();
137 return VStore(atype, alloc);
144 auto s =
VStore{atype, alloc};
153 return bot(env.extends_abstract_dom(), alloc);
160 return top(env.extends_abstract_dom(), alloc);
174 if(is_at_bot) {
return false; }
175 for(
int i = 0; i <
vars(); ++i) {
184 template <
class Alloc = allocator_type>
189 template <
class Alloc>
191 while(snap.size() < data.size()) {
194 is_at_bot.meet_bot();
195 for(
int i = 0; i < snap.size(); ++i) {
196 data[i].join(snap[i]);
197 is_at_bot.join(data[i].
is_bot());
203 template <
bool diagnose,
class F,
class Env,
class Alloc2>
204 CUDA NI
bool interpret_existential(
const F& f, Env& env, tell_type<Alloc2>& tell,
IDiagnostics& diagnostics)
const {
207 if(local_universe::template interpret_tell<diagnose>(f, env, k.dom, diagnostics)) {
208 if(env.interpret(f.map_atype(atype), k.avar, diagnostics)) {
217 template <
bool diagnose,
class F,
class Env,
class Alloc2>
218 CUDA NI
bool interpret_zero_predicate(
const F& f,
const Env& env, tell_type<Alloc2>& tell, IDiagnostics& diagnostics)
const {
222 else if(f.is_false()) {
223 tell.push_back(var_dom<Alloc2>(AVar{}, U::bot()));
232 template <IKind kind,
bool diagnose,
class F,
class Env,
class Alloc2>
233 CUDA NI
bool interpret_unary_predicate(
const F& f,
const Env& env, tell_type<Alloc2>& tell, IDiagnostics& diagnostics)
const {
235 bool res = local_universe::template interpret<kind, diagnose>(f, env, u, diagnostics);
237 const auto& varf =
var_in(f);
241 tell.push_back(var_dom<Alloc2>(varf.v(), u));
244 auto var =
var_in(f, env);
245 if(!var.has_value()) {
248 auto avar = var->get().avar_of(atype);
249 if(!avar.has_value()) {
250 RETURN_INTERPRETATION_ERROR(
"The variable was not declared in the current abstract element (but exists in other abstract elements).");
252 tell.push_back(var_dom<Alloc2>(*avar, u));
261 template <IKind kind,
bool diagnose,
class F,
class Env,
class Alloc2>
262 CUDA NI
bool interpret_predicate(
const F& f, Env& env, tell_type<Alloc2>& tell, IDiagnostics& diagnostics)
const {
263 if(f.type() !=
UNTYPED && f.type() !=
aty()) {
268 return interpret_existential<diagnose>(f, env, tell, diagnostics);
272 case 0:
return interpret_zero_predicate<diagnose>(f, env, tell, diagnostics);
273 case 1:
return interpret_unary_predicate<kind, diagnose>(f, env, tell, diagnostics);
279 template <IKind kind,
bool diagnose = false,
class F,
class Env,
class I>
281 if(f.is_untyped() || f.type() ==
aty()) {
282 return interpret_predicate<kind, diagnose>(f, env, intermediate, diagnostics);
300 template <
bool diagnose = false,
class F,
class Env,
class Alloc2>
302 return interpret<IKind::TELL, diagnose>(f, env, tell, diagnostics);
306 template <
bool diagnose = false,
class F,
class Env,
class Alloc2>
308 return const_cast<this_type*
>(
this)->interpret<IKind::ASK, diagnose>(f,
const_cast<Env&
>(env),
ask, diagnostics);
311 template <
class Group,
class Store>
312 CUDA
void copy_to(Group& group, Store& store)
const {
313 assert(
vars() == store.vars());
314 if(group.thread_rank() == 0) {
315 store.is_at_bot = is_at_bot;
320 for (
int i = group.thread_rank(); i < store.vars(); i += group.num_threads()) {
321 store.data[i] = data[i];
326 void prefetch(
int dstDevice)
const {
328 cudaMemPrefetchAsync(data.data(), data.size() *
sizeof(
universe_type), dstDevice);
335 data = store_type(data.size(), alloc);
338 template <
class Univ>
345 assert(x.
vid() < data.size());
346 return data[x.
vid()];
359 is_at_bot.join_top();
367 assert(x < data.size());
368 bool has_changed = data[x].meet(dom);
369 if(has_changed && data[x].
is_bot()) {
370 is_at_bot.join_top();
384 template <
class Alloc2>
389 if(t[0].avar ==
AVar{}) {
390 return is_at_bot.join(
local::B(
true));
393 for(
int i = 0; i < t.size(); ++i) {
394 largest_vid = battery::max(largest_vid, t[i].avar.vid());
396 if(largest_vid >= data.size()) {
397 data.resize(largest_vid+1);
399 bool has_changed =
false;
400 for(
int i = 0; i < t.size(); ++i) {
401 has_changed |=
embed(t[i].avar, t[i].dom);
407 template <
class U2,
class Alloc2>
409 bool has_changed = is_at_bot.join(other.is_at_bot);
410 int min_size = battery::min(
vars(), other.
vars());
411 for(
int i = 0; i < min_size; ++i) {
412 has_changed |= data[i].meet(other[i]);
414 for(
int i = min_size; i < other.
vars(); ++i) {
415 assert(other[i].
is_top());
421 is_at_bot.meet_bot();
422 for(
int i = 0; i < data.size(); ++i) {
428 template <
class U2,
class Alloc2>
433 int min_size = battery::min(
vars(), other.
vars());
434 bool has_changed = is_at_bot.meet(other.is_at_bot);
435 for(
int i = 0; i < min_size; ++i) {
436 has_changed |= data[i].join(other[i]);
438 for(
int i = min_size; i <
vars(); ++i) {
439 has_changed |= data[i].join(U::top());
441 for(
int i = min_size; i < other.
vars(); ++i) {
442 assert(other[i].
is_top());
450 template <
class Alloc2>
452 for(
int i = 0; i < t.size(); ++i) {
453 if(!(data[t[i].avar.vid()] <= t[i].dom)) {
466 template<
class ExtractionStrategy = NonAtomicExtraction>
467 CUDA
bool is_extractable(
const ExtractionStrategy& strategy = ExtractionStrategy())
const {
471 if constexpr(ExtractionStrategy::atoms) {
472 for(
int i = 0; i < data.size(); ++i) {
473 if(data[i].lb().value() != data[i].ub().value()) {
482 template<
class ExtractionStrategy = NonAtomicExtraction>
483 __device__
bool is_extractable(
auto& group,
const ExtractionStrategy& strategy = ExtractionStrategy())
const {
487 if constexpr(ExtractionStrategy::atoms) {
489 if(group.thread_rank() == 0) {
493 for(
int i = group.thread_rank(); i < data.size(); i += group.num_threads()) {
494 if(data[i].lb().value() != data[i].ub().value()) {
510 template<
class U2,
class Alloc2>
512 if((
void*)&ua != (
void*)
this) {
514 ua.is_at_bot.meet_bot();
519 template<
class U2,
class Env,
class Allocator2>
521 auto f = dom.deinterpret(avar, env, allocator);
528 template<
class Env,
class Allocator2 =
typename Env::allocator_type>
531 if(data.size() == 0) {
532 return is_bot() ? F::make_false() : F::make_true();
534 typename F::Sequence seq{allocator};
535 for(
int i = 0; i < data.size(); ++i) {
537 seq.push_back(F::make_exists(
aty(), env.name_of(v), env.sort_of(v)));
538 seq.push_back(deinterpret(
AVar(
aty(), i), data[i], env, allocator));
540 return F::make_nary(
AND, std::move(seq),
aty());
543 template<
class I,
class Env,
class Allocator2 =
typename Env::allocator_type>
546 if(intermediate.size() == 0) {
547 return F::make_true();
549 else if(intermediate.size() == 1) {
550 return deinterpret(intermediate[0].avar, intermediate[0].dom, env, allocator);
553 typename F::Sequence seq{allocator};
554 for(
int i = 0; i < intermediate.size(); ++i) {
555 seq.push_back(deinterpret(intermediate[i].avar, intermediate[i].dom, env, allocator));
557 return F::make_nary(
AND, std::move(seq),
aty());
566 for(
int i = 0; i <
vars(); ++i) {
568 printf(
"%s", (i+1 ==
vars() ?
"" :
", "));
578 template<
class L,
class K,
class Alloc>
581 using U = decltype(
fmeet(a[0], b[0]));
585 int max_size = battery::max(a.
vars(), b.
vars());
586 int min_size = battery::min(a.
vars(), b.
vars());
588 for(
int i = 0; i < min_size; ++i) {
591 for(
int i = min_size; i < a.
vars(); ++i) {
594 for(
int i = min_size; i < b.
vars(); ++i) {
600 template<
class L,
class K,
class Alloc>
603 using U = decltype(
fjoin(a[0], b[0]));
616 int min_size = battery::min(a.
vars(), b.
vars());
618 for(
int i = 0; i < min_size; ++i) {
625 template<
class L,
class K,
class Alloc1,
class Alloc2>
626 CUDA
bool operator<=(
const VStore<L, Alloc1>& a,
const VStore<K, Alloc2>& b)
632 int min_size = battery::min(a.vars(), b.vars());
633 for(
int i = 0; i < min_size; ++i) {
634 if(!(a[i] <= b[i])) {
638 for(
int i = min_size; i < b.vars(); ++i) {
647 template<
class L,
class K,
class Alloc1,
class Alloc2>
654 int min_size = battery::min(a.
vars(), b.
vars());
656 for(
int i = 0; i < a.
vars(); ++i) {
661 else if(!(a[i] <= b[i])) {
665 else if(!a[i].is_top()) {
670 for(
int i = min_size; i < b.
vars(); ++i) {
679 template<
class L,
class K,
class Alloc1,
class Alloc2>
680 CUDA
bool operator>=(
const VStore<L, Alloc1>& a,
const VStore<K, Alloc2>& b)
685 template<
class L,
class K,
class Alloc1,
class Alloc2>
691 template<
class L,
class K,
class Alloc1,
class Alloc2>
701 int min_size = battery::min(a.
vars(), b.
vars());
702 for(
int i = 0; i < min_size; ++i) {
707 for(
int i = min_size; i < a.
vars(); ++i) {
712 for(
int i = min_size; i < b.
vars(); ++i) {
721 template<
class L,
class K,
class Alloc1,
class Alloc2>
722 CUDA
bool operator!=(
const VStore<L, Alloc1>& a,
const VStore<K, Alloc2>& b)
727 template<
class L,
class Alloc>
734 for(
int i = 0; i < vstore.
vars(); ++i) {
735 s << vstore[i] << (i+1 == vstore.
vars() ?
"" :
", ");
constexpr CUDA int aty() const
Definition: ast.hpp:65
constexpr CUDA int vid() const
Definition: ast.hpp:69
Definition: abstract_deps.hpp:28
Definition: diagnostics.hpp:19
Definition: vstore.hpp:38
battery::vector< var_dom< Alloc >, Alloc > tell_type
Definition: vstore.hpp:58
static CUDA this_type top(AType atype=UNTYPED, const allocator_type &alloc=allocator_type{})
Definition: vstore.hpp:134
CUDA int vars() const
Definition: vstore.hpp:130
CUDA bool deduce(const tell_type< Alloc2 > &t)
Definition: vstore.hpp:385
CUDA local::B is_top() const
Definition: vstore.hpp:173
CUDA bool join(const VStore< U2, Alloc2 > &other)
Definition: vstore.hpp:429
CUDA bool meet(const VStore< U2, Alloc2 > &other)
Definition: vstore.hpp:408
CUDA void copy_to(Group &group, Store &store) const
Definition: vstore.hpp:312
static CUDA this_type bot(AType atype=UNTYPED, const allocator_type &alloc=allocator_type{})
Definition: vstore.hpp:141
CUDA AType aty() const
Definition: vstore.hpp:125
CUDA VStore(AType atype, const allocator_type &alloc=allocator_type())
Definition: vstore.hpp:94
constexpr static const bool is_totally_ordered
Definition: vstore.hpp:68
CUDA int num_deductions() const
Definition: vstore.hpp:460
tell_type< Alloc > ask_type
Definition: vstore.hpp:61
CUDA bool is_extractable(const ExtractionStrategy &strategy=ExtractionStrategy()) const
Definition: vstore.hpp:467
CUDA void print() const
Definition: vstore.hpp:561
static CUDA this_type top(Env &env, const allocator_type &alloc=allocator_type{})
Definition: vstore.hpp:157
CUDA allocator_type get_allocator() const
Definition: vstore.hpp:121
CUDA const universe_type & project(AVar x) const
Definition: vstore.hpp:343
CUDA local::B ask(const ask_type< Alloc2 > &t) const
Definition: vstore.hpp:451
constexpr static const bool preserve_join
Definition: vstore.hpp:71
constexpr static const bool preserve_bot
Definition: vstore.hpp:69
CUDA void project(AVar x, Univ &u) const
Definition: vstore.hpp:339
CUDA const universe_type & operator[](int x) const
Definition: vstore.hpp:350
CUDA void join_top()
Definition: vstore.hpp:420
CUDA bool embed(AVar x, const universe_type &dom)
Definition: vstore.hpp:376
battery::vector< local_universe, Alloc > snapshot_type
Definition: vstore.hpp:64
CUDA local::B is_bot() const
Definition: vstore.hpp:166
U universe_type
Definition: vstore.hpp:40
CUDA NI TFormula< Allocator2 > deinterpret(const Env &env, const Allocator2 &allocator=Allocator2()) const
Definition: vstore.hpp:529
CUDA VStore(const VStore< R, allocator_type > &other)
Definition: vstore.hpp:103
typename universe_type::local_type local_universe
Definition: vstore.hpp:41
friend class VStore
Definition: vstore.hpp:78
CUDA bool embed(int x, const universe_type &dom)
Definition: vstore.hpp:366
CUDA NI bool interpret(const F &f, Env &env, I &intermediate, IDiagnostics &diagnostics) const
Definition: vstore.hpp:280
CUDA VStore(AType atype, int size, const allocator_type &alloc=allocator_type())
Definition: vstore.hpp:98
CUDA void meet_bot()
Definition: vstore.hpp:358
CUDA NI bool interpret_ask(const F &f, const Env &env, ask_type< Alloc2 > &ask, IDiagnostics &diagnostics) const
Definition: vstore.hpp:307
CUDA VStore(this_type &&other)
Definition: vstore.hpp:118
CUDA VStore(const this_type &other)
Definition: vstore.hpp:89
constexpr static const bool preserve_meet
Definition: vstore.hpp:72
CUDA this_type & restore(const snapshot_type< Alloc > &snap)
Definition: vstore.hpp:190
CUDA local::B deduce(int) const
Definition: vstore.hpp:461
constexpr static const char * name
Definition: vstore.hpp:75
CUDA void reset_data(allocator_type alloc)
Definition: vstore.hpp:334
constexpr static const bool preserve_concrete_covers
Definition: vstore.hpp:74
CUDA NI TFormula< Allocator2 > deinterpret(const I &intermediate, const Env &env, const Allocator2 &allocator=Allocator2()) const
Definition: vstore.hpp:544
CUDA void extract(VStore< U2, Alloc2 > &ua) const
Definition: vstore.hpp:511
constexpr static const bool preserve_top
Definition: vstore.hpp:70
CUDA VStore(const VStore< R, Alloc2 > &other, const allocator_type &alloc=allocator_type())
Definition: vstore.hpp:108
constexpr static const bool injective_concretization
Definition: vstore.hpp:73
constexpr static const bool sequential
Definition: vstore.hpp:67
CUDA const universe_type & operator[](AVar x) const
Definition: vstore.hpp:354
CUDA VStore(const VStore< R, Alloc2 > &other, const AbstractDeps< Allocators... > &deps)
Definition: vstore.hpp:115
CUDA snapshot_type< Alloc > snapshot(const Alloc &alloc=Alloc()) const
Definition: vstore.hpp:185
Allocator allocator_type
Definition: vstore.hpp:42
CUDA NI bool interpret_tell(const F &f, Env &env, tell_type< Alloc2 > &tell, IDiagnostics &diagnostics) const
Definition: vstore.hpp:301
static CUDA this_type bot(Env &env, const allocator_type &alloc=allocator_type{})
Definition: vstore.hpp:150
constexpr static const bool is_abstract_universe
Definition: vstore.hpp:66
#define RETURN_INTERPRETATION_ERROR(MSG)
Definition: diagnostics.hpp:155
Definition: abstract_deps.hpp:14
CUDA const TFormula< Allocator, ExtendedSig > & var_in(const TFormula< Allocator, ExtendedSig > &f)
Definition: algorithm.hpp:145
CUDA NI void map_avar_to_lvar(F &f, const Env &env, bool erase_type=false)
Definition: algorithm.hpp:416
constexpr CUDA bool operator==(const CartesianProduct< As... > &a, const CartesianProduct< Bs... > &b)
Definition: cartesian_product.hpp:504
@ AND
Unary arithmetic function symbols.
Definition: ast.hpp:114
constexpr CUDA auto fjoin(const CartesianProduct< As... > &a, const CartesianProduct< Bs... > &b)
Definition: cartesian_product.hpp:464
int AType
Definition: sort.hpp:18
std::ostream & operator<<(std::ostream &s, const CartesianProduct< A, As... > &cp)
Definition: cartesian_product.hpp:531
CUDA NI int num_vars(const F &f)
Definition: algorithm.hpp:153
constexpr CUDA bool operator<(const CartesianProduct< As... > &a, const CartesianProduct< Bs... > &b)
Definition: cartesian_product.hpp:485
constexpr CUDA auto fmeet(const CartesianProduct< As... > &a, const CartesianProduct< Bs... > &b)
Definition: cartesian_product.hpp:471
constexpr CUDA bool operator>(const CartesianProduct< As... > &a, const CartesianProduct< Bs... > &b)
Definition: cartesian_product.hpp:498
#define UNTYPED
Definition: sort.hpp:21
Definition: vstore.hpp:46
CUDA var_dom(const Alloc &)
Definition: vstore.hpp:51
var_dom(const var_dom< Alloc > &)=default
CUDA var_dom(AVar avar, const local_universe &dom)
Definition: vstore.hpp:52
AVar avar
Definition: vstore.hpp:47
local_universe dom
Definition: vstore.hpp:48
CUDA var_dom(const VarDom &other)
Definition: vstore.hpp:54