未验证 提交 6bab3343 编写于 作者: C Chen Weihang 提交者: GitHub

normalize autotune tests dir (#47726)

上级 a4a9ce0e
if(WITH_GPU)
nv_test(
gpu_timer_test
SRCS gpu_timer_test.cu
DEPS gtest)
nv_test(
auto_tune_test
SRCS auto_tune_test.cu
DEPS gtest)
elseif(WITH_ROCM)
hip_test(
gpu_timer_test
SRCS gpu_timer_test.cu
DEPS gtest)
hip_test(
auto_tune_test
SRCS auto_tune_test.cu
DEPS gtest)
endif()
cc_library(cache SRCS cache.cc)
cc_library(
switch_autotune
SRCS switch_autotune.cc
DEPS cache flags)
cc_test(
cache_test
SRCS cache_test.cc
DEPS gtest cache)
......@@ -66,3 +66,28 @@ cc_test(
test_transfer_layout_dev_api
SRCS test_transfer_layout_dev_api.cc
DEPS phi phi_api_utils)
if(WITH_GPU)
nv_test(
test_gpu_timer
SRCS test_gpu_timer.cu
DEPS gtest)
nv_test(
test_auto_tune
SRCS test_auto_tune.cu
DEPS gtest)
elseif(WITH_ROCM)
hip_test(
test_gpu_timer
SRCS test_gpu_timer.cu
DEPS gtest)
hip_test(
test_auto_tune
SRCS test_auto_tune.cu
DEPS gtest)
endif()
cc_test(
test_cache
SRCS test_cache.cc
DEPS gtest cache)
......@@ -12,13 +12,13 @@
// See the License for the specific language governing permissions and
// limitations under the License.
#include "paddle/phi/kernels/autotune/cache.h"
#include <gtest/gtest.h>
#include <cmath>
#include <functional>
#include "paddle/phi/kernels/autotune/cache.h"
#include "glog/logging.h"
enum ConvAlgos { GEMMKernel = 0, CuDNNKernel_1 = 1, CuDNNKernel_2 = 2 };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册