Commit 01490e32 authored by Paul's avatar Paul
Browse files

Update macro

parent a61a8ffc
......@@ -284,7 +284,7 @@ else()
endif()
if(HAS_ROCBLAS_BETA_FEATURES_API)
target_compile_definitions(migraphx_gpu PUBLIC -DROCBLAS_BETA_FEATURES_API -DROCBLAS_NO_DEPRECATED_WARNINGS)
target_compile_definitions(migraphx_gpu PUBLIC -DMIGRAPHX_USE_ROCBLAS_TUNING_API -DROCBLAS_BETA_FEATURES_API -DROCBLAS_NO_DEPRECATED_WARNINGS)
message(STATUS "MIGraphx is using Beta API of rocBLAS")
else()
message(STATUS "rocBLAS does not have Beta API")
......
......@@ -251,7 +251,7 @@ struct gemm_impl
}
}
#ifdef ROCBLAS_BETA_FEATURES_API
#ifdef MIGRAPHX_USE_ROCBLAS_TUNING_API
auto validate(context& ctx, const std::vector<shape>& input_shapes, int32_t solution_idx) const
{
// Create dummy arguments for the shapes, and call the overloaded method
......@@ -380,7 +380,7 @@ struct gemm_impl
ldd,
compute_type);
}
#ifdef ROCBLAS_BETA_FEATURES_API
#ifdef MIGRAPHX_USE_ROCBLAS_TUNING_API
/**
* Find best rocBLAS solution: Get list of solutions and try them all, returning the index
* of the fastest one.
......@@ -557,7 +557,7 @@ int32_t gemm_finalize(context& ctx,
bool compute_fp32,
int32_t solution_idx)
{
#ifdef ROCBLAS_BETA_FEATURES_API
#ifdef MIGRAPHX_USE_ROCBLAS_TUNING_API
// This code should be called only if either the environment var.
// MIGRAPHX_ENABLE_GEMM_TUNING, or option --exhaustive-tune, is set
......@@ -597,7 +597,7 @@ int32_t gemm_finalize(context& ctx,
bool compute_fp32,
int32_t solution_idx)
{
#ifdef ROCBLAS_BETA_FEATURES_API
#ifdef MIGRAPHX_USE_ROCBLAS_TUNING_API
// This code should be called only if either the environment var.
// MIGRAPHX_ENABLE_GEMM_TUNING, or option --exhaustive-tune, is set
......
......@@ -140,7 +140,7 @@ struct rocblas_gemm
void finalize(context& ctx, const shape& output_shape, const std::vector<shape>& input_shapes)
{
#ifdef ROCBLAS_BETA_FEATURES_API
#ifdef MIGRAPHX_USE_ROCBLAS_TUNING_API
if(enabled(MIGRAPHX_ENABLE_GEMM_TUNING{}) or ctx.get_exhaustive_tune_flag())
{
if(this->name() == "gpu::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