mirror of
https://github.com/amd/blis.git
synced 2026-07-14 11:07:11 +00:00
eee3fe1b54
Optionally enable parallelism inside gtestsuite so that we can
check BLIS functions perform correctly when nested parallelism
is in operation. Enable with:
cmake ... -DOPENMP_NESTED={0,1,2,1diff}
where in gtestsuite
- 0 is the default choice with no parallelism.
- 1 and 2 are simple nested parallelism.
- 1diff has one level of parallelism setting different numbers
of threads to be used by BLIS and reference library calls
from each gtestsuite thread.
Note: OMP_NUM_THREADS must be set appropriately to enable or
disable parallelism at each level in the test programs
as desired.
OMP_NUM_THREADS will also define the parallelism used
within the BLIS library (if it is multithreaded), unless
BLIS-specific ways of specifying parallelism have been
used. If a BLIS-specific parallelism option has been set,
the same mechanism will be used in the 1diff option to
vary the number of threads in BLIS per application thread.
AMD-Internal: [CPUPL-3902]
Change-Id: I89f9edb4125c64ef03e025a9f6ccb84960ba8771
103 lines
3.9 KiB
C++
103 lines
3.9 KiB
C++
/*
|
|
|
|
BLIS
|
|
An object-based framework for developing high-performance BLAS-like
|
|
libraries.
|
|
|
|
Copyright (C) 2023 - 2025, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are
|
|
met:
|
|
- Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
- Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
- Neither the name(s) of the copyright holder(s) nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "test_setv.h"
|
|
|
|
class zsetvGeneric :
|
|
public ::testing::TestWithParam<std::tuple<char, gtint_t, gtint_t>> {};
|
|
|
|
GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(zsetvGeneric);
|
|
|
|
TEST_P( zsetvGeneric, API )
|
|
{
|
|
using T = dcomplex;
|
|
//----------------------------------------------------------
|
|
// Initialize values from the parameters passed through
|
|
// test suite instantiation (INSTANTIATE_TEST_SUITE_P).
|
|
//----------------------------------------------------------
|
|
// denotes whether alpha or conjalpha
|
|
char conjalpha = std::get<0>(GetParam());
|
|
// vector length:
|
|
gtint_t n = std::get<1>(GetParam());
|
|
// stride size for x:
|
|
gtint_t incx = std::get<2>(GetParam());
|
|
|
|
T alpha = {1.2, 2.0};
|
|
//----------------------------------------------------------
|
|
// Call generic test body using those parameters
|
|
//----------------------------------------------------------
|
|
|
|
#ifdef OPENMP_NESTED_1diff
|
|
#pragma omp parallel default(shared)
|
|
{
|
|
vary_num_threads();
|
|
//std::cout << "Inside 1diff parallel regions\n";
|
|
test_setv<T>( conjalpha, n, alpha, incx );
|
|
}
|
|
#elif OPENMP_NESTED_2
|
|
#pragma omp parallel default(shared)
|
|
{
|
|
#pragma omp parallel default(shared)
|
|
{
|
|
//std::cout << "Inside 2 parallel regions\n";
|
|
test_setv<T>( conjalpha, n, alpha, incx );
|
|
}
|
|
}
|
|
#elif OPENMP_NESTED_1
|
|
#pragma omp parallel default(shared)
|
|
{
|
|
//std::cout << "Inside 1 parallel region\n";
|
|
test_setv<T>( conjalpha, n, alpha, incx );
|
|
}
|
|
#else
|
|
//std::cout << "Not inside parallel region\n";
|
|
test_setv<T>( conjalpha, n, alpha, incx );
|
|
#endif
|
|
}
|
|
|
|
#ifdef TEST_BLIS_TYPED
|
|
// Black box testing.
|
|
INSTANTIATE_TEST_SUITE_P(
|
|
Blackbox,
|
|
zsetvGeneric,
|
|
::testing::Combine(
|
|
::testing::Values('n','c'), // n: not transpose for x, c: conjugate for x
|
|
::testing::Range(gtint_t(10), gtint_t(101), 10), // m size of vector takes values from 10 to 100 with step size of 10.
|
|
::testing::Values(gtint_t(1)) // stride size for x
|
|
),
|
|
::setvGenericPrint()
|
|
);
|
|
#endif
|