mirror of
https://github.com/exaloop/codon.git
synced 2025-06-03 15:03:52 +08:00
166e1ad455
* Add JIT engine * More engine updates * Fix takeModule() * Parser JIT support [wip] * Fix class definition * Parser JIT support [wip] * Parser JIT support [wip] * Fix LLVM context conflict * Parser JIT support [wip] * Fix JIT engine * Parser JIT support [wip] * Doc fix * JIT fix * Fix JIT exceptions * Refactor * Add JIT status codes * Add compiler class * Better logging * Better logging * Update .gitignore * Add Jupyter Xeus support * Update JIT * Remove print * Update errors * Fix assert * Fix asserts * Update docgen signature * Update file system interface * Update plugin errors * Use numeric_limits * JIT refactor [wip] * Set JIT flag on LLVMVisitor * Update module opt * JIT imports * First Jupyter integration * Update JIT API to return outputs as string * Capture runtime exception output * Fix libbacktrace build * Initial Jupyter support * Format * Fix print * Support run-mode backtraces * Fix multithreaded backtrace * Update backtraces * Upgrade OpenMP * Add libunwind * Fix build * Fix build * Fix build * Fix build * Fix OpenMP & tests * Use libbacktrace instead of libunwind * Add debug listener * Remove unused include * Remove unused class * Fix backtraces * Update backtrace config * Fix debug info generation * Refactor backtraces * Fix ASAN flag * Fix JIT * Fix JIT backtraces * Fix JIT backtrace * Fix Jupyter, fix xeus build flags * Fix JIT output capture * Fix Jupyter * Fix Jupyter Python support * Add __repr_pretty__ support * Update JIT output capturing * Better backtrace method names * Support plugins in JIT mode Co-authored-by: Ibrahim Numanagić <ibrahimpasa@gmail.com>
108 lines
3.4 KiB
C++
108 lines
3.4 KiB
C++
#include "test.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "codon/sir/util/matching.h"
|
|
|
|
using namespace codon::ir;
|
|
|
|
TEST_F(SIRCoreTest, FuncRealizationAndVarInsertionEraseAndIterators) {
|
|
auto *fn = module->Nr<BodiedFunc>();
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
auto *fnType = module->unsafeGetFuncType("**test_type**", module->getIntType(),
|
|
{module->getIntType()});
|
|
std::vector<std::string> names = {"foo"};
|
|
fn->realize(cast<types::FuncType>(fnType), names);
|
|
ASSERT_TRUE(fn->isGlobal());
|
|
|
|
ASSERT_EQ(1, std::distance(fn->arg_begin(), fn->arg_end()));
|
|
ASSERT_EQ(module->getIntType(), fn->arg_front()->getType());
|
|
|
|
auto *var = module->Nr<Var>(module->getIntType(), false, "hi");
|
|
fn->push_back(var);
|
|
ASSERT_EQ(1, std::distance(fn->begin(), fn->end()));
|
|
fn->erase(fn->begin());
|
|
ASSERT_EQ(0, std::distance(fn->begin(), fn->end()));
|
|
fn->insert(fn->begin(), var);
|
|
ASSERT_EQ(1, std::distance(fn->begin(), fn->end()));
|
|
ASSERT_EQ(module->getIntType(), fn->front()->getType());
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, BodiedFuncQueryAndReplace) {
|
|
auto *fn = module->Nr<BodiedFunc>();
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
fn->setJIT();
|
|
ASSERT_TRUE(fn->isJIT());
|
|
|
|
auto *body = fn->getBody();
|
|
ASSERT_FALSE(body);
|
|
ASSERT_EQ(0, fn->getUsedValues().size());
|
|
|
|
body = module->Nr<SeriesFlow>();
|
|
fn->setBody(body);
|
|
ASSERT_EQ(body, fn->getBody());
|
|
|
|
auto used = fn->getUsedValues();
|
|
ASSERT_EQ(1, used.size());
|
|
ASSERT_EQ(body, used[0]);
|
|
|
|
ASSERT_EQ(1, fn->replaceUsedValue(body, module->Nr<SeriesFlow>()));
|
|
ASSERT_DEATH(fn->replaceUsedValue(fn->getBody(), module->Nr<VarValue>(nullptr)), "");
|
|
ASSERT_NE(fn->getBody(), body);
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, BodiedFuncUnmangledName) {
|
|
auto *fn = module->Nr<BodiedFunc>("Int.foo");
|
|
fn->setUnmangledName("foo");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
ASSERT_EQ("foo", fn->getUnmangledName());
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, BodiedFuncCloning) {
|
|
auto *fn = module->Nr<BodiedFunc>("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setJIT();
|
|
fn->setBody(module->Nr<SeriesFlow>());
|
|
ASSERT_TRUE(util::match(fn, cv->clone(fn)));
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, ExternalFuncUnmangledNameAndCloning) {
|
|
auto *fn = module->Nr<ExternalFunc>("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setUnmangledName("foo");
|
|
ASSERT_EQ("foo", fn->getUnmangledName());
|
|
ASSERT_TRUE(util::match(fn, cv->clone(fn)));
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, InternalFuncParentTypeUnmangledNameAndCloning) {
|
|
auto *fn = module->Nr<InternalFunc>("fn.1");
|
|
fn->setUnmangledName("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setParentType(module->getIntType());
|
|
ASSERT_EQ("fn", fn->getUnmangledName());
|
|
ASSERT_EQ(fn->getParentType(), module->getIntType());
|
|
ASSERT_TRUE(util::match(fn, cv->clone(fn)));
|
|
}
|
|
|
|
TEST_F(SIRCoreTest, LLVMFuncUnmangledNameQueryAndReplace) {
|
|
auto *fn = module->Nr<LLVMFunc>("fn");
|
|
fn->realize(module->unsafeGetDummyFuncType(), {});
|
|
|
|
fn->setLLVMBody("body");
|
|
fn->setLLVMDeclarations("decl");
|
|
|
|
std::vector<types::Generic> literals = {types::Generic(1),
|
|
types::Generic(module->getIntType())};
|
|
fn->setLLVMLiterals(literals);
|
|
|
|
ASSERT_EQ("body", fn->getLLVMBody());
|
|
ASSERT_EQ("decl", fn->getLLVMDeclarations());
|
|
|
|
ASSERT_EQ(1, fn->replaceUsedType(module->getIntType(), module->getFloatType()));
|
|
ASSERT_EQ(module->getFloatType(), fn->literal_back().getTypeValue());
|
|
}
|