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

Update macro

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