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
c07c4dfc
"docker/vscode:/vscode.git/clone" did not exist on "9a6b81ef7d3ae032e501ec09ca4a7dc17e31d3e2"
Commit
c07c4dfc
authored
Apr 20, 2020
by
Jing Zhang
Browse files
add dot2 emulate
parent
f328497e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
26 additions
and
0 deletions
+26
-0
composable_kernel/include/utility/float_type.nvidia.hpp.in
composable_kernel/include/utility/float_type.nvidia.hpp.in
+26
-0
No files found.
composable_kernel/include/utility/float_type.nvidia.hpp.in
View file @
c07c4dfc
...
@@ -150,6 +150,30 @@ struct inner_product_with_conversion
...
@@ -150,6 +150,30 @@ struct inner_product_with_conversion
__device__ T operator()(float a, float b) const { return convert(a) * convert(b); }
__device__ T operator()(float a, float b) const { return convert(a) * convert(b); }
#if 0
__device__ T operator()(half2_t a, half2_t b) const
{
const auto p_a_half = reinterpret_cast<const float*>(&a);
const auto p_b_half = reinterpret_cast<const float*>(&b);
T acc = convert(p_a_half[0]) * convert(p_b_half[0]);
return acc;
}
__device__ T operator()(half4_t a, half4_t b) const
{
const auto p_a_half = reinterpret_cast<const float*>(&a);
const auto p_b_half = reinterpret_cast<const float*>(&b);
T acc = 0;
for(index_t v = 0; v < 2; ++v)
{
acc += convert(p_a_half[v]) * convert(p_b_half[v]);
}
return acc;
}
#else
__device__ T operator()(half2_t a, half2_t b) const
__device__ T operator()(half2_t a, half2_t b) const
{
{
const half* p_a_half = reinterpret_cast<const half*>(&a);
const half* p_a_half = reinterpret_cast<const half*>(&a);
...
@@ -177,6 +201,8 @@ struct inner_product_with_conversion
...
@@ -177,6 +201,8 @@ struct inner_product_with_conversion
return acc;
return acc;
}
}
#endif
};
};
} // namespace ck
} // namespace ck
...
...
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