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
b256a3b0
Commit
b256a3b0
authored
May 25, 2023
by
Paul
Browse files
Format
parent
96e062d1
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
6 deletions
+9
-6
src/targets/gpu/include/migraphx/gpu/compile_hip_code_object.hpp
...gets/gpu/include/migraphx/gpu/compile_hip_code_object.hpp
+3
-3
src/targets/gpu/jit/ck_gemm.cpp
src/targets/gpu/jit/ck_gemm.cpp
+6
-3
No files found.
src/targets/gpu/include/migraphx/gpu/compile_hip_code_object.hpp
View file @
b256a3b0
...
@@ -42,9 +42,9 @@ struct hip_compile_options
...
@@ -42,9 +42,9 @@ struct hip_compile_options
std
::
size_t
local
;
std
::
size_t
local
;
std
::
vector
<
shape
>
inputs
;
std
::
vector
<
shape
>
inputs
;
shape
output
;
shape
output
;
std
::
string
kernel_name
=
"kernel"
;
std
::
string
kernel_name
=
"kernel"
;
std
::
string
params
=
""
;
std
::
string
params
=
""
;
std
::
vector
<
shape
>
virtual_inputs
=
{};
std
::
vector
<
shape
>
virtual_inputs
=
{};
std
::
vector
<
src_file
>
additional_src_files
=
{};
std
::
vector
<
src_file
>
additional_src_files
=
{};
/**
/**
...
...
src/targets/gpu/jit/ck_gemm.cpp
View file @
b256a3b0
...
@@ -291,9 +291,12 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
...
@@ -291,9 +291,12 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
const
auto
block_size
=
solution
.
block_size
;
const
auto
block_size
=
solution
.
block_size
;
hip_compile_options
options
;
hip_compile_options
options
;
std
::
transform
(
ck_headers
.
begin
(),
ck_headers
.
end
(),
std
::
back_inserter
(
options
.
additional_src_files
),
[
&
](
auto
&&
p
)
{
std
::
transform
(
ck_headers
.
begin
(),
return
src_file
{
fs
::
path
{
p
.
first
},
p
.
second
};
ck_headers
.
end
(),
});
std
::
back_inserter
(
options
.
additional_src_files
),
[
&
](
auto
&&
p
)
{
return
src_file
{
fs
::
path
{
p
.
first
},
p
.
second
};
});
auto
grid_size
=
can_fold_batch
?
blocks_per_batch
:
batch_count
*
blocks_per_batch
;
auto
grid_size
=
can_fold_batch
?
blocks_per_batch
:
batch_count
*
blocks_per_batch
;
options
.
set_launch_params
(
v
,
grid_size
*
block_size
,
block_size
);
options
.
set_launch_params
(
v
,
grid_size
*
block_size
,
block_size
);
options
.
inputs
=
inputs
;
options
.
inputs
=
inputs
;
...
...
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