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

Method generate

src/driver/main.cpp:508–537  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

506 }
507
508 auto generate(const program& p,
509 const target& t,
510 bool offload,
511 unsigned batch,
512 dims_map map_input_dims = {})
513 {
514 parameter_map m;
515 auto param_shapes = p.get_parameter_shapes();
516 std::unordered_map<std::string, shape> static_param_shapes;
517 for(auto&& param : param_shapes)
518 {
519 if(contains(map_input_dims, param.first))
520 static_param_shapes[param.first] = {param.second.type(),
521 map_input_dims[param.first]};
522 else
523 static_param_shapes[param.first] = param.second.to_static(batch);
524 }
525
526 for(auto&& s : fill0)
527 m[s] = fill_argument(static_param_shapes.at(s), 0);
528 for(auto&& s : fill1)
529 m[s] = fill_argument(static_param_shapes.at(s), 1);
530 fill_param_map(m, static_param_shapes, t, offload);
531 auto load_arg_map = program_params::parse_load_args(load_args_info, t, offload);
532 for(auto&& arg : load_arg_map)
533 {
534 m[arg.first] = arg.second;
535 }
536 return m;
537 }
538};
539
540struct compiler_target

Callers 3

paramsMethod · 0.45
host_paramsMethod · 0.45
runMethod · 0.45

Calls 7

containsFunction · 0.85
fill_argumentFunction · 0.85
fill_param_mapFunction · 0.85
to_staticMethod · 0.80
atMethod · 0.80
get_parameter_shapesMethod · 0.45
typeMethod · 0.45

Tested by

no test coverage detected