Skip to content

GitLab

"sgl-router/git@developer.sourcefind.cn:change/sglang.git" did not exist on "957482c8f2675c90aaa133e187f5d6e00af8ceae"
Unverified Commit 98ca0a18 authored by Umang Yadav's avatar Umang Yadav Committed by GitHub
Browse files

Merge branch 'develop' into rm_elim_workspace

parents b688bf6a c0398ded
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
src/targets/gpu/include/migraphx/gpu/miopen.hpp
View file @ 98ca0a18
...@@ -11,9 +11,10 @@ ...@@ -11,9 +11,10 @@
#include <sstream> #include <sstream>
#ifdef HAS_FIND_MODE_API #ifdef MIGRAPHX_HAS_FIND_MODE_API
extern "C" miopenStatus_t miopenHiddenSetConvolutionFindMode(miopenConvolutionDescriptor_t convDesc, extern "C" miopenStatus_t
int findMode); miopenHiddenSetConvolutionFindMode(miopenConvolutionDescriptor_t convDesc, // NOLINT
int findMode); // NOLINT
#endif #endif
namespace migraphx { namespace migraphx {
...@@ -104,7 +105,7 @@ inline convolution_descriptor make_conv(const T& op) ...@@ -104,7 +105,7 @@ inline convolution_descriptor make_conv(const T& op)
c.get(), padding.size(), padding.data(), stride.data(), dilation.data(), c_mode); c.get(), padding.size(), padding.data(), stride.data(), dilation.data(), c_mode);
if(op.group > 1) if(op.group > 1)
miopenSetConvolutionGroupCount(c.get(), op.group); miopenSetConvolutionGroupCount(c.get(), op.group);
#ifdef HAS_FIND_MODE_API #ifdef MIGRAPHX_HAS_FIND_MODE_API
miopenHiddenSetConvolutionFindMode(c.get(), 1); // Normal mode miopenHiddenSetConvolutionFindMode(c.get(), 1); // Normal mode
#endif #endif
return c; return c;
......
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