"docs/source/Tutorial/Nnictl.rst" did not exist on "9f32a06f16774fe2226f6c4b9b0da014fbcb77c1"
Commit d7b1895a authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

Merge branch 'develop' of https://github.com/ROCmSoftwarePlatform/AMDMIGraphX into nhwc_workaround

parents 1add453a a83371ca
...@@ -261,11 +261,13 @@ auto compute_op(rank<1>, ...@@ -261,11 +261,13 @@ auto compute_op(rank<1>,
template <class T, class F> template <class T, class F>
argument compute_op(rank<0>, argument compute_op(rank<0>,
const T& x, const T& x,
const shape&, const shape& output,
const std::vector<argument>&, const std::vector<argument>& inputs,
const std::vector<module_ref>&, const std::vector<module_ref>& module_args,
F) F)
{ {
if(module_args.empty())
return compute_op(x, output, inputs);
std::string name = x.name(); std::string name = x.name();
MIGRAPHX_THROW("Not computable: " + name); MIGRAPHX_THROW("Not computable: " + name);
} }
......
...@@ -45,6 +45,8 @@ ...@@ -45,6 +45,8 @@
namespace migraphx { namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS { inline namespace MIGRAPHX_INLINE_NS {
struct value;
#ifdef DOXYGEN #ifdef DOXYGEN
/// An interface for a compilation target /// An interface for a compilation target
...@@ -123,28 +125,41 @@ supported_segments target_find_supported(T&, const_module_ref, support_metric) ...@@ -123,28 +125,41 @@ supported_segments target_find_supported(T&, const_module_ref, support_metric)
} }
<% <%
interface('target', interface('target',
virtual('name', returns='std::string', const=True), virtual('name', returns = 'std::string', const = True),
virtual('get_passes', ctx='context&', options='const compile_options&', returns='std::vector<pass>', const=True), virtual('get_passes',
virtual('get_context', returns='context', const=True), ctx = 'context&',
virtual('find_supported', returns='supported_segments', mod='const_module_ref', m='support_metric', const=True, default='target_find_supported'), options = 'const compile_options&',
virtual('copy_to', returns = 'std::vector<pass>',
returns = 'argument', const = True),
input = 'const argument&', virtual('get_context', returns = 'context', const = True),
const = True, virtual('find_supported',
default = 'copy_to_target'), returns = 'supported_segments',
virtual('copy_from', mod = 'const_module_ref',
returns = 'argument', m = 'support_metric',
input = 'const argument&', const = True,
const = True, default = 'target_find_supported'),
default = 'copy_from_target'), virtual('copy_to',
virtual('allocate', s='const shape&', returns='argument', const=True, returns = 'argument',
default = 'target_allocate') input = 'const argument&',
) const = True,
%> default = 'copy_to_target'),
virtual('copy_from',
returns = 'argument',
input = 'const argument&',
const = True,
default = 'copy_from_target'),
virtual('allocate',
s = 'const shape&',
returns = 'argument',
const = True,
default = 'target_allocate')) %>
#endif #endif
void migraphx_to_value(value& v, const target& t);
void migraphx_from_value(const value& v, target& t);
} // namespace MIGRAPHX_INLINE_NS } // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx } // namespace migraphx
......
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