Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
b3e9d8f8
Commit
b3e9d8f8
authored
May 02, 2019
by
Paul
Browse files
Formatting
parent
3ed217c9
Changes
21
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
40 additions
and
44 deletions
+40
-44
src/include/migraphx/op/concat.hpp
src/include/migraphx/op/concat.hpp
+1
-1
src/include/migraphx/op/leaky_relu.hpp
src/include/migraphx/op/leaky_relu.hpp
+2
-3
src/include/migraphx/reflect.hpp
src/include/migraphx/reflect.hpp
+5
-4
src/targets/cpu/lowering.cpp
src/targets/cpu/lowering.cpp
+12
-17
src/targets/gpu/include/migraphx/gpu/abs.hpp
src/targets/gpu/include/migraphx/gpu/abs.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/batchnorm.hpp
src/targets/gpu/include/migraphx/gpu/batchnorm.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/concat.hpp
src/targets/gpu/include/migraphx/gpu/concat.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/contiguous.hpp
src/targets/gpu/include/migraphx/gpu/contiguous.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/elu.hpp
src/targets/gpu/include/migraphx/gpu/elu.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/gather.hpp
src/targets/gpu/include/migraphx/gpu/gather.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/gemm.hpp
src/targets/gpu/include/migraphx/gpu/gemm.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/leaky_relu.hpp
src/targets/gpu/include/migraphx/gpu/leaky_relu.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/logsoftmax.hpp
src/targets/gpu/include/migraphx/gpu/logsoftmax.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/lrn.hpp
src/targets/gpu/include/migraphx/gpu/lrn.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/miopen.hpp
src/targets/gpu/include/migraphx/gpu/miopen.hpp
+5
-4
src/targets/gpu/include/migraphx/gpu/relu.hpp
src/targets/gpu/include/migraphx/gpu/relu.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/sigmoid.hpp
src/targets/gpu/include/migraphx/gpu/sigmoid.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/softmax.hpp
src/targets/gpu/include/migraphx/gpu/softmax.hpp
+1
-1
src/targets/gpu/include/migraphx/gpu/tanh.hpp
src/targets/gpu/include/migraphx/gpu/tanh.hpp
+1
-1
test/eliminate_allocation_test.cpp
test/eliminate_allocation_test.cpp
+1
-1
No files found.
src/include/migraphx/op/concat.hpp
View file @
b3e9d8f8
...
@@ -25,7 +25,7 @@ struct concat
...
@@ -25,7 +25,7 @@ struct concat
{
{
return
pack
(
f
(
self
.
axis
,
"axis"
));
return
pack
(
f
(
self
.
axis
,
"axis"
));
}
}
std
::
string
name
()
const
{
return
"concat"
;
}
std
::
string
name
()
const
{
return
"concat"
;
}
std
::
vector
<
std
::
size_t
>
compute_offsets
(
const
shape
&
output_shape
,
std
::
vector
<
std
::
size_t
>
compute_offsets
(
const
shape
&
output_shape
,
const
std
::
vector
<
argument
>&
args
)
const
const
std
::
vector
<
argument
>&
args
)
const
...
...
src/include/migraphx/op/leaky_relu.hpp
View file @
b3e9d8f8
...
@@ -19,20 +19,19 @@ namespace op {
...
@@ -19,20 +19,19 @@ namespace op {
struct
leaky_relu
struct
leaky_relu
{
{
float
alpha
;
float
alpha
;
template
<
class
Self
,
class
F
>
template
<
class
Self
,
class
F
>
static
auto
reflect
(
Self
&
self
,
F
f
)
static
auto
reflect
(
Self
&
self
,
F
f
)
{
{
return
pack
(
f
(
self
.
alpha
,
"alpha"
));
return
pack
(
f
(
self
.
alpha
,
"alpha"
));
}
}
std
::
string
name
()
const
{
return
"leaky_relu"
;
}
std
::
string
name
()
const
{
return
"leaky_relu"
;
}
shape
compute_shape
(
std
::
vector
<
shape
>
inputs
)
const
shape
compute_shape
(
std
::
vector
<
shape
>
inputs
)
const
{
{
check_shapes
{
inputs
,
*
this
}.
has
(
1
);
check_shapes
{
inputs
,
*
this
}.
has
(
1
);
return
inputs
.
front
();
return
inputs
.
front
();
}
}
};
};
}
// namespace op
}
// namespace op
...
...
src/include/migraphx/reflect.hpp
View file @
b3e9d8f8
...
@@ -13,7 +13,7 @@ namespace detail {
...
@@ -13,7 +13,7 @@ namespace detail {
struct
reflect_placeholder
struct
reflect_placeholder
{
{
template
<
class
...
Ts
>
template
<
class
...
Ts
>
int
operator
()(
Ts
&&
...)
const
int
operator
()(
Ts
&&
...)
const
{
{
return
0
;
return
0
;
...
@@ -33,14 +33,15 @@ auto reflect_impl(rank<0>, T&, Selector)
...
@@ -33,14 +33,15 @@ auto reflect_impl(rank<0>, T&, Selector)
}
}
template
<
class
T
>
template
<
class
T
>
auto
reflectable_impl
(
rank
<
1
>
,
T
&&
x
)
->
decltype
(
T
::
reflect
(
x
,
reflect_placeholder
{}),
std
::
true_type
{});
auto
reflectable_impl
(
rank
<
1
>
,
T
&&
x
)
->
decltype
(
T
::
reflect
(
x
,
reflect_placeholder
{}),
std
::
true_type
{});
template
<
class
T
>
template
<
class
T
>
auto
reflectable_impl
(
rank
<
0
>
,
T
&&
)
->
decltype
(
std
::
false_type
{});
auto
reflectable_impl
(
rank
<
0
>
,
T
&&
)
->
decltype
(
std
::
false_type
{});
}
// namespace detail
}
// namespace detail
template
<
class
T
>
template
<
class
T
>
using
is_reflectable
=
decltype
(
detail
::
reflectable_impl
(
rank
<
1
>
{},
std
::
declval
<
T
>
()));
using
is_reflectable
=
decltype
(
detail
::
reflectable_impl
(
rank
<
1
>
{},
std
::
declval
<
T
>
()));
template
<
class
T
,
class
Selector
>
template
<
class
T
,
class
Selector
>
...
...
src/targets/cpu/lowering.cpp
View file @
b3e9d8f8
...
@@ -354,29 +354,23 @@ struct cpu_op
...
@@ -354,29 +354,23 @@ struct cpu_op
{
{
return
op
.
compute
(
output_shape
,
std
::
move
(
args
));
return
op
.
compute
(
output_shape
,
std
::
move
(
args
));
}
}
friend
bool
operator
==
(
const
cpu_op
&
x
,
const
cpu_op
&
y
)
friend
bool
operator
==
(
const
cpu_op
&
x
,
const
cpu_op
&
y
)
{
return
x
.
op
==
y
.
op
;
}
{
return
x
.
op
==
y
.
op
;
}
friend
bool
operator
==
(
const
cpu_op
&
x
,
const
operation
&
y
)
friend
bool
operator
==
(
const
cpu_op
&
x
,
const
operation
&
y
)
{
{
if
(
x
.
name
()
!=
y
.
name
())
if
(
x
.
name
()
!=
y
.
name
())
return
false
;
return
false
;
return
x
==
any_cast
<
cpu_op
>
(
y
);
return
x
==
any_cast
<
cpu_op
>
(
y
);
}
}
friend
bool
operator
==
(
const
operation
&
x
,
const
cpu_op
&
y
)
friend
bool
operator
==
(
const
operation
&
x
,
const
cpu_op
&
y
)
{
return
y
==
x
;
}
{
return
y
==
x
;
}
};
};
// struct cpu_contiguous
// struct cpu_contiguous
// {
// {
// op::contiguous op;
// op::contiguous op;
// std::string name() const { return "cpu::contiguous"; }
// std::string name() const { return "cpu::contiguous"; }
// shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); }
// shape compute_shape(const std::vector<shape>& inputs) const { return
// argument compute(context&, const shape& output_shape, std::vector<argument> args) const
// op.compute_shape(inputs); } argument compute(context&, const shape& output_shape,
// std::vector<argument> args) const
// {
// {
// return op.compute(output_shape, std::move(args));
// return op.compute(output_shape, std::move(args));
// }
// }
...
@@ -419,8 +413,9 @@ struct cpu_pad
...
@@ -419,8 +413,9 @@ struct cpu_pad
// {
// {
// op::concat op;
// op::concat op;
// std::string name() const { return "cpu::concat"; }
// std::string name() const { return "cpu::concat"; }
// shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); }
// shape compute_shape(const std::vector<shape>& inputs) const { return
// argument compute(context&, const shape& output_shape, std::vector<argument> args) const
// op.compute_shape(inputs); } argument compute(context&, const shape& output_shape,
// std::vector<argument> args) const
// {
// {
// return op.compute(output_shape, std::move(args));
// return op.compute(output_shape, std::move(args));
// }
// }
...
@@ -481,7 +476,8 @@ struct cpu_gemm
...
@@ -481,7 +476,8 @@ struct cpu_gemm
// {
// {
// op::gather op;
// op::gather op;
// std::string name() const { return "cpu::gather"; }
// std::string name() const { return "cpu::gather"; }
// shape compute_shape(const std::vector<shape>& inputs) const { return op.compute_shape(inputs); }
// shape compute_shape(const std::vector<shape>& inputs) const { return
// op.compute_shape(inputs); }
// argument compute(context&, const shape& output_shape, std::vector<argument> args) const
// argument compute(context&, const shape& output_shape, std::vector<argument> args) const
// {
// {
...
@@ -903,7 +899,7 @@ struct cpu_apply
...
@@ -903,7 +899,7 @@ struct cpu_apply
apply_map
[
"leaky_relu"
]
=
extend_op
<
cpu_unary
<
leaky_relu_op
>
,
op
::
leaky_relu
>
();
apply_map
[
"leaky_relu"
]
=
extend_op
<
cpu_unary
<
leaky_relu_op
>
,
op
::
leaky_relu
>
();
apply_map
[
"logsoftmax"
]
=
extend_op
<
cpu_logsoftmax
,
op
::
logsoftmax
>
();
apply_map
[
"logsoftmax"
]
=
extend_op
<
cpu_logsoftmax
,
op
::
logsoftmax
>
();
apply_map
[
"elu"
]
=
extend_op
<
cpu_unary
<
elu_op
>
,
op
::
elu
>
();
apply_map
[
"elu"
]
=
extend_op
<
cpu_unary
<
elu_op
>
,
op
::
elu
>
();
apply_map
[
"softmax"
]
=
simple_op
<
softmax2d
>
();
apply_map
[
"softmax"
]
=
simple_op
<
softmax2d
>
();
apply_map
[
"pad"
]
=
extend_op
<
cpu_pad
,
op
::
pad
>
();
apply_map
[
"pad"
]
=
extend_op
<
cpu_pad
,
op
::
pad
>
();
// apply_map["contiguous"] = extend_op<cpu_contiguous, op::contiguous>();
// apply_map["contiguous"] = extend_op<cpu_contiguous, op::contiguous>();
// apply_map["concat"] = extend_op<cpu_concat, op::concat>();
// apply_map["concat"] = extend_op<cpu_concat, op::concat>();
...
@@ -930,7 +926,6 @@ struct cpu_apply
...
@@ -930,7 +926,6 @@ struct cpu_apply
// apply_map["div"] = simple_op<cpu_binary<div_op>>();
// apply_map["div"] = simple_op<cpu_binary<div_op>>();
// apply_map["max"] = simple_op<cpu_binary<max_op>>();
// apply_map["max"] = simple_op<cpu_binary<max_op>>();
// apply_map["min"] = simple_op<cpu_binary<min_op>>();
// apply_map["min"] = simple_op<cpu_binary<min_op>>();
}
}
void
apply
()
void
apply
()
...
@@ -946,7 +941,7 @@ struct cpu_apply
...
@@ -946,7 +941,7 @@ struct cpu_apply
{
{
apply_map
.
at
(
it
->
name
())(
it
);
apply_map
.
at
(
it
->
name
())(
it
);
}
}
else
if
(
is_context_free
(
it
->
get_operator
()))
else
if
(
is_context_free
(
it
->
get_operator
()))
{
{
apply_cpu_op
(
it
);
apply_cpu_op
(
it
);
}
}
...
...
src/targets/gpu/include/migraphx/gpu/abs.hpp
View file @
b3e9d8f8
...
@@ -13,7 +13,7 @@ struct context;
...
@@ -13,7 +13,7 @@ struct context;
struct
miopen_abs
struct
miopen_abs
{
{
shared
<
activation_descriptor
>
ad
;
shared
<
activation_descriptor
>
ad
;
template
<
class
Self
,
class
F
>
template
<
class
Self
,
class
F
>
static
auto
reflect
(
Self
&
self
,
F
f
)
static
auto
reflect
(
Self
&
self
,
F
f
)
{
{
...
...
src/targets/gpu/include/migraphx/gpu/batchnorm.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_batch_norm_inference
...
@@ -19,7 +19,7 @@ struct miopen_batch_norm_inference
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::batch_norm_inference"
;
}
std
::
string
name
()
const
{
return
"gpu::batch_norm_inference"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/concat.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct hip_concat
...
@@ -19,7 +19,7 @@ struct hip_concat
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::concat"
;
}
std
::
string
name
()
const
{
return
"gpu::concat"
;
}
shape
compute_shape
(
std
::
vector
<
shape
>
inputs
)
const
;
shape
compute_shape
(
std
::
vector
<
shape
>
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/contiguous.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_contiguous
...
@@ -19,7 +19,7 @@ struct miopen_contiguous
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::contiguous"
;
}
std
::
string
name
()
const
{
return
"gpu::contiguous"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
compute
(
context
&
,
shape
output_shape
,
const
std
::
vector
<
argument
>&
args
)
const
;
argument
compute
(
context
&
,
shape
output_shape
,
const
std
::
vector
<
argument
>&
args
)
const
;
...
...
src/targets/gpu/include/migraphx/gpu/elu.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_elu
...
@@ -19,7 +19,7 @@ struct miopen_elu
{
{
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::elu"
;
}
std
::
string
name
()
const
{
return
"gpu::elu"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/gather.hpp
View file @
b3e9d8f8
...
@@ -20,7 +20,7 @@ struct hip_gather
...
@@ -20,7 +20,7 @@ struct hip_gather
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::gather"
;
}
std
::
string
name
()
const
{
return
"gpu::gather"
;
}
shape
compute_shape
(
std
::
vector
<
shape
>
inputs
)
const
;
shape
compute_shape
(
std
::
vector
<
shape
>
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/gemm.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_gemm
...
@@ -19,7 +19,7 @@ struct miopen_gemm
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::gemm"
;
}
std
::
string
name
()
const
{
return
"gpu::gemm"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/leaky_relu.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_leaky_relu
...
@@ -19,7 +19,7 @@ struct miopen_leaky_relu
{
{
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::leaky_relu"
;
}
std
::
string
name
()
const
{
return
"gpu::leaky_relu"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/logsoftmax.hpp
View file @
b3e9d8f8
...
@@ -31,7 +31,7 @@ struct hip_logsoftmax
...
@@ -31,7 +31,7 @@ struct hip_logsoftmax
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::logsoftmax"
;
}
std
::
string
name
()
const
{
return
"gpu::logsoftmax"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/lrn.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_lrn
...
@@ -19,7 +19,7 @@ struct miopen_lrn
{
{
return
gpu
::
reflect
(
self
.
ldesc
.
get
(),
f
);
return
gpu
::
reflect
(
self
.
ldesc
.
get
(),
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::lrn"
;
}
std
::
string
name
()
const
{
return
"gpu::lrn"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/miopen.hpp
View file @
b3e9d8f8
...
@@ -162,7 +162,7 @@ inline fused_operator_args make_fused_args()
...
@@ -162,7 +162,7 @@ inline fused_operator_args make_fused_args()
return
make_obj
<
fused_operator_args
>
(
&
miopenCreateOperatorArgs
);
return
make_obj
<
fused_operator_args
>
(
&
miopenCreateOperatorArgs
);
}
}
template
<
class
F
>
template
<
class
F
>
auto
reflect
(
miopenActivationDescriptor_t
ad
,
F
f
)
auto
reflect
(
miopenActivationDescriptor_t
ad
,
F
f
)
{
{
miopenActivationMode_t
mode
;
miopenActivationMode_t
mode
;
...
@@ -173,15 +173,16 @@ auto reflect(miopenActivationDescriptor_t ad, F f)
...
@@ -173,15 +173,16 @@ auto reflect(miopenActivationDescriptor_t ad, F f)
return
pack
(
f
(
mode
,
"mode"
),
f
(
alpha
,
"alpha"
),
f
(
beta
,
"beta"
),
f
(
gamma
,
"gamma"
));
return
pack
(
f
(
mode
,
"mode"
),
f
(
alpha
,
"alpha"
),
f
(
beta
,
"beta"
),
f
(
gamma
,
"gamma"
));
}
}
template
<
class
F
>
template
<
class
F
>
auto
reflect
(
miopenLRNDescriptor_t
lrnd
,
F
f
)
auto
reflect
(
miopenLRNDescriptor_t
lrnd
,
F
f
)
{
{
miopenLRNMode_t
mode
;;
miopenLRNMode_t
mode
;
;
unsigned
int
n
;
unsigned
int
n
;
double
alpha
;
double
alpha
;
double
beta
;
double
beta
;
double
k
;
double
k
;
miopenGetLRNDescriptor
(
lrnd
,
&
mode
,
&
n
,
&
alpha
,
&
beta
,
&
k
);
miopenGetLRNDescriptor
(
lrnd
,
&
mode
,
&
n
,
&
alpha
,
&
beta
,
&
k
);
return
pack
(
f
(
mode
,
"mode"
),
f
(
n
,
"n"
),
f
(
alpha
,
"alpha"
),
f
(
beta
,
"beta"
),
f
(
k
,
"k"
));
return
pack
(
f
(
mode
,
"mode"
),
f
(
n
,
"n"
),
f
(
alpha
,
"alpha"
),
f
(
beta
,
"beta"
),
f
(
k
,
"k"
));
}
}
...
...
src/targets/gpu/include/migraphx/gpu/relu.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_relu
...
@@ -19,7 +19,7 @@ struct miopen_relu
{
{
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::relu"
;
}
std
::
string
name
()
const
{
return
"gpu::relu"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/sigmoid.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_sigmoid
...
@@ -19,7 +19,7 @@ struct miopen_sigmoid
{
{
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::sigmoid"
;
}
std
::
string
name
()
const
{
return
"gpu::sigmoid"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/softmax.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_softmax
...
@@ -19,7 +19,7 @@ struct miopen_softmax
{
{
return
migraphx
::
reflect
(
self
.
op
,
f
);
return
migraphx
::
reflect
(
self
.
op
,
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::softmax"
;
}
std
::
string
name
()
const
{
return
"gpu::softmax"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
src/targets/gpu/include/migraphx/gpu/tanh.hpp
View file @
b3e9d8f8
...
@@ -19,7 +19,7 @@ struct miopen_tanh
...
@@ -19,7 +19,7 @@ struct miopen_tanh
{
{
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
return
gpu
::
reflect
(
self
.
ad
.
get
(),
f
);
}
}
std
::
string
name
()
const
{
return
"gpu::tanh"
;
}
std
::
string
name
()
const
{
return
"gpu::tanh"
;
}
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
shape
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
;
argument
argument
...
...
test/eliminate_allocation_test.cpp
View file @
b3e9d8f8
...
@@ -26,7 +26,7 @@ struct allocate
...
@@ -26,7 +26,7 @@ struct allocate
{
{
return
migraphx
::
pack
(
f
(
self
.
s
,
"shape"
));
return
migraphx
::
pack
(
f
(
self
.
s
,
"shape"
));
}
}
std
::
string
name
()
const
{
return
"allocate"
;
}
std
::
string
name
()
const
{
return
"allocate"
;
}
migraphx
::
shape
compute_shape
(
const
std
::
vector
<
migraphx
::
shape
>&
inputs
)
const
migraphx
::
shape
compute_shape
(
const
std
::
vector
<
migraphx
::
shape
>&
inputs
)
const
{
{
...
...
Prev
1
2
Next
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