MCPcopy Create free account
hub / github.com/ROCm/AMDMIGraphX / create_program

Method create_program

test/verify/test_softmax_large1.cpp:33–42  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31struct test_softmax_large1 : verify_program<test_softmax_large1>
32{
33 migraphx::program create_program() const
34 {
35 migraphx::program p;
36 auto* mm = p.get_main_module();
37 auto x = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, {2, 4}});
38 auto large = mm->add_literal({migraphx::shape{migraphx::shape::float_type}, {10000}});
39 auto add = migraphx::add_common_op(*mm, migraphx::make_op("add"), {x, large});
40 mm->add_instruction(migraphx::make_op("softmax", {{"axis", -1}}), add);
41 return p;
42 }
43
44 std::string section() const { return "reduce"; }
45};

Callers

nothing calls this directly

Calls 6

add_common_opFunction · 0.85
get_main_moduleMethod · 0.80
add_parameterMethod · 0.80
make_opFunction · 0.50
add_literalMethod · 0.45
add_instructionMethod · 0.45

Tested by

no test coverage detected