"convert/git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "0ce8bcfdaa4ddb58ba1c46c4e9fed1faec315bd6"
Commit 9d171dc5 authored by Astha Rai's avatar Astha Rai
Browse files

small edits to 3d transpose profiler

parent a36dff02
...@@ -19,22 +19,14 @@ void add_device_transpose_f16_instances( ...@@ -19,22 +19,14 @@ void add_device_transpose_f16_instances(
std::vector<std::unique_ptr<DeviceElementwise<ck::Tuple<F16>, ck::Tuple<F16>, PassThrough, 5>>>& std::vector<std::unique_ptr<DeviceElementwise<ck::Tuple<F16>, ck::Tuple<F16>, PassThrough, 5>>>&
instances) instances)
{ {
#ifdef CK_ENABLE_FP16
add_device_operation_instances(instances, device_transpose_f16_instances{}); add_device_operation_instances(instances, device_transpose_f16_instances{});
#else
ignore = instances;
#endif
} }
void add_device_transpose_f32_instances( void add_device_transpose_f32_instances(
std::vector<std::unique_ptr<DeviceElementwise<ck::Tuple<F32>, ck::Tuple<F32>, PassThrough, 5>>>& std::vector<std::unique_ptr<DeviceElementwise<ck::Tuple<F32>, ck::Tuple<F32>, PassThrough, 5>>>&
instances) instances)
{ {
#ifdef CK_ENABLE_FP32
add_device_operation_instances(instances, device_transpose_f32_instances{}); add_device_operation_instances(instances, device_transpose_f32_instances{});
#else
ignore = instances;
#endif
} }
} // namespace instance } // namespace instance
......
...@@ -82,4 +82,4 @@ int profile_transpose(int argc, char* argv[]) ...@@ -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);
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment