iframe-proxy | Sunbelt Computer Software

History

112 lines (99 loc) · 3.89 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

* The MIT License (MIT)

* Permission is hereby granted, free of charge, to any person obtaining a copy

* of this software and associated documentation files (the "Software"), to deal

* in the Software without restriction, including without limitation the rights

* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

* copies of the Software, and to permit persons to whom the Software is

* furnished to do so, subject to the following conditions:

* The above copyright notice and this permission notice shall be included in

* all copies or substantial portions of the Software.

* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

* THE SOFTWARE.

#include <migraphx/compile_src.hpp>

#include <migraphx/dynamic_loader.hpp>

#include <migraphx/fileutils.hpp>

#include <migraphx/cpp_generator.hpp>

#include <migraphx/module.hpp>

#include <migraphx/make_op.hpp>

#include <test.hpp>

// NOLINTNEXTLINE

const std::string_view add_42_src = R"migraphx(

EXPORT extern "C" int add(int x)

{

return x+42;

}

)migraphx";

// NOLINTNEXTLINE

const std::string_view preamble = R"migraphx(

#include <cmath>

)migraphx";

template <class F>

static std::function<F> compile_function(std::string_view src, const std::string& symbol_name)

{

migraphx::src_compiler compiler;

compiler.flags.emplace_back("-std=c++14");

#ifndef _WIN32

compiler.flags.emplace_back("-fPIC");

compiler.flags.emplace_back("-DEXPORT=\"\"");

#else

compiler.flags.emplace_back("-DEXPORT=__declspec(dllexport)");

#endif

compiler.flags.emplace_back("-shared");

compiler.output = migraphx::make_shared_object_filename("simple");

migraphx::src_file f{"main.cpp", src};

auto image = compiler.compile({f});

return migraphx::dynamic_loader{image}.get_function<F>(symbol_name);

}

template <class F>

static std::function<F> compile_module(const migraphx::module& m)

{

migraphx::cpp_generator g;

g.fmap([](auto&& name) { return "std::" + name; });

g.create_function(g.generate_module(m).set_attributes({"EXPORT extern \"C\""}));

return compile_function<F>(g.str().insert(0, preamble), m.name());

}

TEST_CASE(simple_run)

{

auto f = compile_function<int(int)>(add_42_src, "add");

EXPECT(f(8) == 50);

EXPECT(f(10) == 52);

}

TEST_CASE(generate_module)

{

migraphx::module m("foo");

auto x = m.add_parameter("x", migraphx::shape::float_type);

auto y = m.add_parameter("y", migraphx::shape::float_type);

auto sum = m.add_instruction(migraphx::make_op("add"), x, y);

m.add_instruction(migraphx::make_op("sqrt"), sum);

auto f = compile_module<float(float, float)>(m);

EXPECT(test::within_abs(f(2, 2), 2));

EXPECT(test::within_abs(f(10, 6), 4));

EXPECT(test::within_abs(f(1, 2), std::sqrt(3)));

}

TEST_CASE(generate_module_with_literals)

{

migraphx::module m("foo");

auto x = m.add_parameter("x", migraphx::shape::float_type);

auto y = m.add_parameter("y", migraphx::shape::float_type);

auto z = m.add_literal(1.f);

auto sum1 = m.add_instruction(migraphx::make_op("add"), x, z);

auto sum2 = m.add_instruction(migraphx::make_op("add"), sum1, y);

m.add_instruction(migraphx::make_op("sqrt"), sum2);

auto f = compile_module<float(float, float)>(m);

EXPECT(test::within_abs(f(1, 2), 2));

EXPECT(test::within_abs(f(9, 6), 4));

EXPECT(test::within_abs(f(0, 2), std::sqrt(3)));

}

int main(int argc, const char* argv[]) { test::run(argc, argv); }

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

jit.cpp

Latest commit

History

jit.cpp

File metadata and controls