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
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