Commit e3ac3847 authored by charlie's avatar charlie
Browse files

test multiple module dependency

parent 6e78a8cc
......@@ -7,6 +7,7 @@
#include <sstream>
#include "test.hpp"
#include <migraphx/make_op.hpp>
#include <migraphx/verify.hpp>
#include <basic_ops.hpp>
......@@ -312,4 +313,16 @@ TEST_CASE(module_without_bypass)
EXPECT(found);
}
TEST_CASE(multiple_module_dependency)
{
// Test when an instruction from a submodule depends on previous module
migraphx::program p;
auto* mm = p.get_main_module();
auto* sub = p.create_module("sub");
auto l1 = mm->add_literal(migraphx::literal(3));
sub->add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {2, 2 ,3}}}), l1);
p.compile(migraphx::ref::target{});
p.eval({});
}
int main(int argc, const char* argv[]) { test::run(argc, argv); }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment