Commit cad8ceed authored by Paul's avatar Paul
Browse files

Add env var to disable ck_gemm

parent 1a05cce8
......@@ -3,10 +3,14 @@
#include <migraphx/pass_manager.hpp>
#include <migraphx/make_op.hpp>
#include <migraphx/register_op.hpp>
#include <migraphx/env.hpp>
namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS {
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_DISABLE_CK_GEMM);
MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_DISABLE_CK_GEMM_FUSION);
struct module;
namespace gpu {
......@@ -116,7 +120,9 @@ struct find_ck_gemm
void fuse_ck::apply(module_pass_manager& mpm) const
{
if (not enabled(MIGRAPHX_DISABLE_CK_GEMM_FUSION{}))
match::find_matches(mpm, find_ck_gemm_pointwise{});
if (not enabled(MIGRAPHX_DISABLE_CK_GEMM{}))
match::find_matches(mpm, find_ck_gemm{});
}
......
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