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
c57b18ea
Commit
c57b18ea
authored
Jan 16, 2022
by
Shucai Xiao
Browse files
add a gpu implementation for the get_tuple_elem op
parent
fc42d852
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
1 deletion
+5
-1
src/include/migraphx/op/get_tuple_elem.hpp
src/include/migraphx/op/get_tuple_elem.hpp
+1
-1
src/targets/gpu/CMakeLists.txt
src/targets/gpu/CMakeLists.txt
+2
-0
src/targets/gpu/lowering.cpp
src/targets/gpu/lowering.cpp
+2
-0
No files found.
src/include/migraphx/op/get_tuple_elem.hpp
View file @
c57b18ea
...
@@ -46,7 +46,7 @@ struct get_tuple_elem
...
@@ -46,7 +46,7 @@ struct get_tuple_elem
return
vec_args
.
at
(
index
);
return
vec_args
.
at
(
index
);
}
}
std
::
ptrdiff_t
output_alias
(
const
std
::
vector
<
shape
>&
)
const
{
return
0
;
}
//
std::ptrdiff_t output_alias(const std::vector<shape>&) const { return 0; }
};
};
}
// namespace op
}
// namespace op
...
...
src/targets/gpu/CMakeLists.txt
View file @
c57b18ea
...
@@ -143,6 +143,7 @@ add_library(migraphx_gpu
...
@@ -143,6 +143,7 @@ add_library(migraphx_gpu
fuse_ops.cpp
fuse_ops.cpp
gather.cpp
gather.cpp
gemm_impl.cpp
gemm_impl.cpp
get_tuple_elem.cpp
hip.cpp
hip.cpp
int8_conv_pack.cpp
int8_conv_pack.cpp
int8_gemm_pack.cpp
int8_gemm_pack.cpp
...
@@ -200,6 +201,7 @@ register_migraphx_gpu_ops(hip_
...
@@ -200,6 +201,7 @@ register_migraphx_gpu_ops(hip_
exp
exp
floor
floor
gather
gather
get_tuple_elem
greater
greater
less
less
log
log
...
...
src/targets/gpu/lowering.cpp
View file @
c57b18ea
...
@@ -27,6 +27,7 @@
...
@@ -27,6 +27,7 @@
#include <migraphx/gpu/elu.hpp>
#include <migraphx/gpu/elu.hpp>
#include <migraphx/gpu/equal.hpp>
#include <migraphx/gpu/equal.hpp>
#include <migraphx/gpu/gemm.hpp>
#include <migraphx/gpu/gemm.hpp>
#include <migraphx/gpu/get_tuple_elem.hpp>
#include <migraphx/gpu/greater.hpp>
#include <migraphx/gpu/greater.hpp>
#include <migraphx/gpu/int8_conv_pack.hpp>
#include <migraphx/gpu/int8_conv_pack.hpp>
#include <migraphx/gpu/leaky_relu.hpp>
#include <migraphx/gpu/leaky_relu.hpp>
...
@@ -162,6 +163,7 @@ struct miopen_apply
...
@@ -162,6 +163,7 @@ struct miopen_apply
add_extend_op
(
"convert"
);
add_extend_op
(
"convert"
);
add_extend_op
(
"elu"
);
add_extend_op
(
"elu"
);
add_extend_op
(
"gather"
);
add_extend_op
(
"gather"
);
add_extend_op
(
"get_tuple_elem"
);
add_extend_op
(
"leaky_relu"
);
add_extend_op
(
"leaky_relu"
);
add_extend_op
(
"logsoftmax"
);
add_extend_op
(
"logsoftmax"
);
add_extend_op
(
"lrn"
);
add_extend_op
(
"lrn"
);
...
...
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