"docs/vscode:/vscode.git/clone" did not exist on "bf7daa8fabbc4a8ba24df7b0e9822f8035ab0429"
Commit 647d5dc5 authored by Paul's avatar Paul
Browse files

Format

parent dd93e13c
...@@ -91,7 +91,7 @@ struct softmax_compiler : compiler<softmax_compiler> ...@@ -91,7 +91,7 @@ struct softmax_compiler : compiler<softmax_compiler>
options.inputs = inputs; options.inputs = inputs;
options.kernel_name = "softmax_kernel"; options.kernel_name = "softmax_kernel";
if (enabled(MIGRAPHX_USE_FAST_SOFTMAX{})) if(enabled(MIGRAPHX_USE_FAST_SOFTMAX{}))
options.params = "-DMIGRAPHX_USE_FAST_SOFTMAX"; options.params = "-DMIGRAPHX_USE_FAST_SOFTMAX";
auto src = interpolate_string( auto src = interpolate_string(
......
...@@ -34,7 +34,7 @@ __device__ void softmax(Input input, Output output) ...@@ -34,7 +34,7 @@ __device__ void softmax(Input input, Output output)
{ {
reduce::block::run<reduce::with_axis<Input, Axis>>([&](auto, auto r) { reduce::block::run<reduce::with_axis<Input, Axis>>([&](auto, auto r) {
#ifdef MIGRAPHX_USE_FAST_SOFTMAX #ifdef MIGRAPHX_USE_FAST_SOFTMAX
const auto c = vec_at(r.slice(input)[0], 0); const auto c = vec_at(r.slice(input)[0], 0);
#else #else
const auto c = r.reduce(op::max{}, lowest{}, op::id{})(input); const auto c = r.reduce(op::max{}, lowest{}, op::id{})(input);
#endif #endif
......
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