Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
MIGraphX
Commits
01490e32
Commit
01490e32
authored
Oct 18, 2023
by
Paul
Browse files
Update macro
parent
a61a8ffc
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
6 additions
and
6 deletions
+6
-6
src/targets/gpu/CMakeLists.txt
src/targets/gpu/CMakeLists.txt
+1
-1
src/targets/gpu/gemm_impl.cpp
src/targets/gpu/gemm_impl.cpp
+4
-4
src/targets/gpu/include/migraphx/gpu/gemm.hpp
src/targets/gpu/include/migraphx/gpu/gemm.hpp
+1
-1
No files found.
src/targets/gpu/CMakeLists.txt
View file @
01490e32
...
...
@@ -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"
)
...
...
src/targets/gpu/gemm_impl.cpp
View file @
01490e32
...
...
@@ -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
...
...
src/targets/gpu/include/migraphx/gpu/gemm.hpp
View file @
01490e32
...
...
@@ -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"
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment