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
a6e8172c
Commit
a6e8172c
authored
Aug 16, 2018
by
Paul
Browse files
Formatting
parent
5aa0ba49
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
19 additions
and
18 deletions
+19
-18
src/include/migraph/generate.hpp
src/include/migraph/generate.hpp
+12
-12
src/program.cpp
src/program.cpp
+2
-2
src/targets/gpu/eliminate_workspace.cpp
src/targets/gpu/eliminate_workspace.cpp
+2
-2
src/targets/gpu/lowering.cpp
src/targets/gpu/lowering.cpp
+3
-2
No files found.
src/include/migraph/generate.hpp
View file @
a6e8172c
...
@@ -7,12 +7,12 @@
...
@@ -7,12 +7,12 @@
namespace
migraph
{
namespace
migraph
{
template
<
class
T
>
template
<
class
T
>
struct
xorshf96_generator
struct
xorshf96_generator
{
{
unsigned
long
x
=
123456789
;
unsigned
long
x
=
123456789
;
unsigned
long
y
=
362436069
;
unsigned
long
y
=
362436069
;
unsigned
long
z
=
521288629
;
unsigned
long
z
=
521288629
;
constexpr
T
operator
()()
constexpr
T
operator
()()
{
{
...
...
src/program.cpp
View file @
a6e8172c
src/targets/gpu/eliminate_workspace.cpp
View file @
a6e8172c
...
@@ -28,7 +28,7 @@ void eliminate_workspace::apply(program& p) const
...
@@ -28,7 +28,7 @@ void eliminate_workspace::apply(program& p) const
}
}
}
}
auto
ws
=
p
.
add_parameter
(
"workspace"
,
shape
{
shape
::
int8_type
,
{
n
}});
auto
ws
=
p
.
add_parameter
(
"workspace"
,
shape
{
shape
::
int8_type
,
{
n
}});
for
(
auto
&&
a
:
allocs
)
for
(
auto
&&
a
:
allocs
)
{
{
p
.
replace_instruction
(
a
,
ws
);
p
.
replace_instruction
(
a
,
ws
);
p
.
remove_instruction
(
a
);
p
.
remove_instruction
(
a
);
...
...
src/targets/gpu/lowering.cpp
View file @
a6e8172c
...
@@ -125,7 +125,8 @@ struct miopen_convolution
...
@@ -125,7 +125,8 @@ struct miopen_convolution
workspace_size
,
workspace_size
,
false
);
false
);
algo
=
perf
.
fwd_algo
;
algo
=
perf
.
fwd_algo
;
return
algo
==
miopenConvolutionFwdAlgoWinograd
?
shape
{
shape
::
int8_type
,
{
0
}}
:
workspace_shape
;
return
algo
==
miopenConvolutionFwdAlgoWinograd
?
shape
{
shape
::
int8_type
,
{
0
}}
:
workspace_shape
;
}
}
};
};
...
@@ -332,7 +333,7 @@ struct miopen_apply
...
@@ -332,7 +333,7 @@ struct miopen_apply
}
}
}
}
instruction_ref
insert_allocation
(
instruction_ref
ins
,
const
shape
&
s
,
std
::
string
tag
=
""
)
instruction_ref
insert_allocation
(
instruction_ref
ins
,
const
shape
&
s
,
std
::
string
tag
=
""
)
{
{
if
(
ins
==
--
prog
->
end
())
if
(
ins
==
--
prog
->
end
())
{
{
...
...
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