Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
37f4de96
Commit
37f4de96
authored
Oct 03, 2018
by
Paul
Browse files
Add miopen fusion objects
parent
3d912972
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
14 additions
and
1 deletion
+14
-1
requirements.txt
requirements.txt
+1
-1
src/targets/gpu/include/migraph/gpu/miopen.hpp
src/targets/gpu/include/migraph/gpu/miopen.hpp
+13
-0
No files found.
requirements.txt
View file @
37f4de96
google/protobuf
-DCMAKE_POSITION_INDEPENDENT_CODE=On
RadeonOpenCompute/rocm-cmake@d82a77c
--build
ROCmSoftwarePlatform/MIOpen
,http://gitlab1.amd.com/pfultz/miopen/-/archive/57dd8372fa043f4bb500ac297f2c479e1c819e89/miopen-57dd8372fa043f4bb500ac297f2c479e1c819e89.tar.bz2
ROCmSoftwarePlatform/MIOpen
@1.5.0
ROCmSoftwarePlatform/rocBLAS@v14.0.1
blaze,https://bitbucket.org/blaze-lib/blaze/get/f0755dea0e03.tar.gz
-X header -DHEADER_DIR=blaze
src/targets/gpu/include/migraph/gpu/miopen.hpp
View file @
37f4de96
...
...
@@ -17,6 +17,8 @@ using pooling_descriptor = MIGRAPH_MANAGE_PTR(miopenPoolingDescriptor_t,
miopenDestroyPoolingDescriptor
);
using
activation_descriptor
=
MIGRAPH_MANAGE_PTR
(
miopenActivationDescriptor_t
,
miopenDestroyActivationDescriptor
);
using
fusion_plan_descriptor
=
MIGRAPH_MANAGE_PTR
(
miopenFusionPlanDescriptor_t
,
miopenDestroyFusionPlan
);
using
fused_operator_args
=
MIGRAPH_MANAGE_PTR
(
miopenOperatorArgs_t
,
miopenDestroyOperatorArgs
);
template
<
class
Result
,
class
F
,
class
...
Ts
>
Result
make_obj
(
F
f
,
Ts
...
xs
)
...
...
@@ -84,6 +86,17 @@ inline activation_descriptor make_relu()
return
ad
;
}
inline
fusion_plan_descriptor
make_fusion_plan
(
const
migraph
::
shape
&
input
)
{
auto
t
=
make_tensor
(
input
);
return
make_obj
<
fusion_plan_descriptor
>
(
&
miopenCreateFusionPlan
,
miopenVerticalFusion
,
t
.
get
());
}
inline
fused_operator_args
make_fused_args
()
{
return
make_obj
<
fused_operator_args
>
(
&
miopenCreateOperatorArgs
);
}
}
// namespace gpu
}
// namespace migraph
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment