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
composable_kernel
Commits
9d171dc5
"llm/vscode:/vscode.git/clone" did not exist on "440b7190ed2e704290f99c1d8124a9bccd1bcc05"
Commit
9d171dc5
authored
Nov 28, 2023
by
Astha Rai
Browse files
small edits to 3d transpose profiler
parent
a36dff02
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
9 deletions
+1
-9
library/src/tensor_operation_instance/gpu/transpose/device_transpose_instances_3d.cpp
..._instance/gpu/transpose/device_transpose_instances_3d.cpp
+0
-8
profiler/src/profile_transpose.cpp
profiler/src/profile_transpose.cpp
+1
-1
No files found.
library/src/tensor_operation_instance/gpu/transpose/device_transpose_instances_3d.cpp
View file @
9d171dc5
...
...
@@ -19,22 +19,14 @@ void add_device_transpose_f16_instances(
std
::
vector
<
std
::
unique_ptr
<
DeviceElementwise
<
ck
::
Tuple
<
F16
>
,
ck
::
Tuple
<
F16
>
,
PassThrough
,
5
>>>&
instances
)
{
#ifdef CK_ENABLE_FP16
add_device_operation_instances
(
instances
,
device_transpose_f16_instances
{});
#else
ignore
=
instances
;
#endif
}
void
add_device_transpose_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DeviceElementwise
<
ck
::
Tuple
<
F32
>
,
ck
::
Tuple
<
F32
>
,
PassThrough
,
5
>>>&
instances
)
{
#ifdef CK_ENABLE_FP32
add_device_operation_instances
(
instances
,
device_transpose_f32_instances
{});
#else
ignore
=
instances
;
#endif
}
}
// namespace instance
...
...
profiler/src/profile_transpose.cpp
View file @
9d171dc5
...
...
@@ -82,4 +82,4 @@ int profile_transpose(int argc, char* argv[])
}
}
REGISTER_PROFILER_OPERATION
(
OP_NAME
,
OP_DESC
,
profile_
gemm_
transpose
);
REGISTER_PROFILER_OPERATION
(
OP_NAME
,
OP_DESC
,
profile_transpose
);
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