From 12626bcfe413da2c0d980b54fa5f88959f1b6bf9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E4=BE=AF=E5=A5=87?= Date: Thu, 24 Oct 2024 00:56:36 +0800 Subject: [PATCH] Update gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu with include "cutlass/gemm/device/gemm_universal.h" (#1569) fix compile with `cmake .. -DCUTLASS_ENABLE_TESTS=ON -DCUTLASS_TEST_LEVEL=2` --- test/unit/gemm/device/gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/test/unit/gemm/device/gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu b/test/unit/gemm/device/gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu index f94d6d603..29fda22ce 100644 --- a/test/unit/gemm/device/gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu +++ b/test/unit/gemm/device/gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu @@ -37,6 +37,7 @@ #include "../../common/cutlass_unit_test.h" #include "cutlass/cutlass.h" #include "cutlass/gemm/device/gemm.h" +#include "cutlass/gemm/device/gemm_universal.h" #include "cutlass/util/host_tensor.h" #include "cutlass/util/reference/host/gemm.h" #include "cutlass/util/reference/host/tensor_compare.h"