squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
a0d787b746
cutlass/test/unit/gemm/device/CMakeLists.txt

737 lines
21 KiB
CMake
Raw Normal View History

New updates for 2.11 (#775) * New updates. * Minor profiler updates Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2023-01-21 05:32:57 +08:00
# Copyright (c) 2017 - 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
# SPDX-License-Identifier: BSD-3-Clause
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
#
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
#
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
# 1. Redistributions of source code must retain the above copyright notice, this
# list of conditions and the following disclaimer.
#
# 2. 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.
#
# 3. Neither the name of the copyright holder 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
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
add_custom_target(
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
cutlass_test_unit_gemm_device
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
DEPENDS
cutlass_test_unit_gemm_device_simt
cutlass_test_unit_gemm_device_tensorop_sm70
cutlass_test_unit_gemm_device_tensorop_sm75
cutlass_test_unit_gemm_device_tensorop_f16_sm80
cutlass_test_unit_gemm_device_tensorop_f32_sm80
cutlass_test_unit_gemm_device_tensorop_f32_tf32_sm80
cutlass_test_unit_gemm_device_tensorop_f64
cutlass_test_unit_gemm_device_tensorop_s32_sm80
cutlass_test_unit_gemm_device_wmma
cutlass_test_unit_gemm_device_tensorop_planar_complex
cutlass_test_unit_gemm_device_sparse_tensorop_sm80
CUTLASS 2.6 (#298) CUTLASS 2.6
2021-07-23 12:40:53 +08:00
cutlass_test_unit_gemv_device
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
cutlass_test_unit_gemm_device_tensorop_sm90
cutlass_test_unit_gemm_device_tensorop_cluster_multicast_sm90
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
add_custom_target(
test_unit_gemm_device
DEPENDS
test_unit_gemm_device_simt
test_unit_gemm_device_tensorop_sm70
test_unit_gemm_device_tensorop_sm75
test_unit_gemm_device_tensorop_f16_sm80
test_unit_gemm_device_tensorop_f32_sm80
test_unit_gemm_device_tensorop_f32_tf32_sm80
test_unit_gemm_device_tensorop_f64
test_unit_gemm_device_tensorop_s32_sm80
test_unit_gemm_device_wmma
test_unit_gemm_device_tensorop_planar_complex
test_unit_gemm_device_sparse_tensorop_sm80
CUTLASS 2.6 (#298) CUTLASS 2.6
2021-07-23 12:40:53 +08:00
test_unit_gemv_device
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
test_unit_gemm_device_tensorop_sm90
)
add_custom_target(
cutlass_test_unit_gemm_device_sm90
DEPENDS
cutlass_test_unit_gemm_device_tensorop_sm90
cutlass_test_unit_gemm_device_tensorop_cluster_multicast_sm90
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_simt
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.1 (#83) CUTLASS 2.1 contributes: - BLAS-style host-side API added to CUTLASS Library - Planar Complex GEMM kernels targeting Volta and Turing Tensor Cores - Minor enhancements and bug fixes
2020-04-08 04:51:25 +08:00
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_sgemm_nt_sm80.cu
simt_sgemm_tn_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
simt_cgemm_nt_sm80.cu
simt_cgemm_tn_sm80.cu
simt_f8gemm_tn_sm50.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_cgemm_nn_sm50.cu
simt_cgemm_nt_sm50.cu
simt_cgemm_tn_sm50.cu
simt_cgemm_tt_sm50.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.6 (#298) CUTLASS 2.6
2021-07-23 12:40:53 +08:00
simt_qgemm_nn_sm50.cu
simt_qgemm_nt_sm50.cu
simt_qgemm_tn_sm50.cu
simt_qgemm_tt_sm50.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_dgemm_nn_sm50.cu
simt_dgemm_nt_sm50.cu
simt_dgemm_tn_sm50.cu
simt_dgemm_tt_sm50.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_hgemm_nn_sm50.cu
simt_hgemm_nt_sm50.cu
simt_hgemm_tn_sm50.cu
simt_hgemm_tt_sm50.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_igemm_nn_sm50.cu
simt_igemm_nt_sm50.cu
simt_igemm_tn_sm50.cu
simt_igemm_tt_sm50.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_int8_igemm_sm61_sliced_k.cu
simt_int8_igemm_sm61.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_sgemm_nn_sm50.cu
simt_sgemm_nt_sm50.cu
simt_sgemm_tn_sm50.cu
simt_sgemm_tt_sm50.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
simt_zgemm_nn_sm50.cu
simt_zgemm_nt_sm50.cu
simt_zgemm_tn_sm50.cu
simt_zgemm_tt_sm50.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_splitk_simt_sm50.cu
)
CUTLASS 2.2 (#96) Adds support for NVIDIA Ampere Architecture features. CUDA 11 Toolkit recommended.
2020-06-09 07:17:35 +08:00
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_simt_3x
BATCH_SOURCES ON
BATCH_SIZE 4
sm50_gemm_f32_f32_f32_simt.cu
sm80_gemm_f32_f32_f32_simt.cu
sm50_gemm_f64_f64_f64_simt.cu
sm80_gemm_f64_f64_f64_simt.cu
sm61_gemm_s8_s8_s32_simt.cu
)
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_sm70
CUTLASS 2.2 (#96) Adds support for NVIDIA Ampere Architecture features. CUDA 11 Toolkit recommended.
2020-06-09 07:17:35 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.2 (#96) Adds support for NVIDIA Ampere Architecture features. CUDA 11 Toolkit recommended.
2020-06-09 07:17:35 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_f16n_f16n_f32t_volta_tensor_op_f32_sm70.cu
gemm_f16n_f16t_f32t_volta_tensor_op_f32_sm70.cu
gemm_f16t_f16n_f32t_volta_tensor_op_f32_sm70.cu
gemm_f16t_f16t_f32t_volta_tensor_op_f32_sm70.cu
CUTLASS 2.2 (#96) Adds support for NVIDIA Ampere Architecture features. CUDA 11 Toolkit recommended.
2020-06-09 07:17:35 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_f16n_f16n_f16t_volta_tensor_op_f32_sm70.cu
gemm_f16n_f16t_f16t_volta_tensor_op_f16_sm70.cu
gemm_f16t_f16n_f16t_volta_tensor_op_f16_sm70.cu
gemm_splitk_tensor_op_sm70.cu
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_sm75
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.2 (#96) Adds support for NVIDIA Ampere Architecture features. CUDA 11 Toolkit recommended.
2020-06-09 07:17:35 +08:00
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
gemm_universal_f16n_f16t_f32n_tensor_op_f32_sm75.cu
CUTLASS 2.5
2021-02-26 22:58:26 +08:00
gemm_universal_f16n_f16t_f32t_tensor_op_f32_sm75.cu
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
gemm_f16t_f16n_f16t_tensor_op_f16_sm75.cu
gemm_f16n_f16t_f16t_tensor_op_f16_sm75.cu
CUTLASS 2.1 (#83) CUTLASS 2.1 contributes: - BLAS-style host-side API added to CUTLASS Library - Planar Complex GEMM kernels targeting Volta and Turing Tensor Cores - Minor enhancements and bug fixes
2020-04-08 04:51:25 +08:00
gemm_f16n_f16t_f16t_tensor_op_f16_slicedk_sm75.cu
gemm_f16t_f16n_f16t_tensor_op_f16_slicedk_sm75.cu
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
gemm_f16n_f16n_f16t_tensor_op_f32_sm75.cu
gemm_f16n_f16n_f32t_tensor_op_f32_sm75.cu
gemm_f16n_f16t_f32t_tensor_op_f32_sm75.cu
gemm_f16t_f16n_f32t_tensor_op_f32_sm75.cu
gemm_f16t_f16t_f32t_tensor_op_f32_sm75.cu
gemm_f16n_f16n_f32n_tensor_op_f32_sm75.cu
gemm_f16t_f16t_f32n_tensor_op_f32_sm75.cu
gemm_s8n_s8t_s8n_tensor_op_s32_sm75.cu
gemm_s8t_s8n_s32t_tensor_op_s32_sm75.cu
gemm_s8t_s8n_s32n_tensor_op_s32_sm75.cu
gemm_s8t_s8n_s8t_tensor_op_s32_sm75.cu
gemm_s8t_s8n_s8n_tensor_op_s32_sm75.cu
gemm_s4n_s4t_s4n_tensor_op_s32_sm75.cu
gemm_s4t_s4n_s32t_tensor_op_s32_sm75.cu
gemm_s4t_s4n_s32n_tensor_op_s32_sm75.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
gemm_s4t_s4n_s4n_tensor_op_s32_sm75.cu
gemm_s4t_s4n_s4t_tensor_op_s32_sm75.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_b1t_b1n_s32t_tensor_op_s32_sm75.cu
gemm_b1t_b1n_s32n_tensor_op_s32_sm75.cu
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_splitk_serial_tensor_op_sm75.cu
gemm_splitk_tensor_op_sm75.cu
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_f16_sm80
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_f16t_f16n_f16t_tensor_op_f16_slicedk_sm80.cu
gemm_f16n_f16t_f16t_tensor_op_f16_slicedk_sm80.cu
)
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_f32_sm80
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_f16n_f16n_f16t_tensor_op_f32_sm80.cu
gemm_f16n_f16n_f32n_tensor_op_f32_sm80.cu
gemm_f16n_f16n_f32t_tensor_op_f32_sm80.cu
gemm_f16n_f16t_f16t_tensor_op_f16_sm80.cu
gemm_f16n_f16t_f32t_tensor_op_f32_sm80.cu
gemm_f16t_f16n_f16t_tensor_op_f16_sm80.cu
gemm_f16t_f16n_f32t_tensor_op_f32_sm80.cu
gemm_f16t_f16t_f32n_tensor_op_f32_sm80.cu
gemm_f16t_f16t_f32t_tensor_op_f32_sm80.cu
gemm_bf16n_bf16n_f32t_tensor_op_f32_sm80.cu
gemm_bf16t_bf16t_bf16t_tensor_op_f32_sm80.cu
CUTLASS 2.8 (#363) CUTLASS 2.8
2021-11-20 05:26:35 +08:00
gemm_f16n_f16n_f16n_direct_store_tensor_op_f32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_f32_sm80_3x
sm80_gemm_s8_s8_s32_tensor_op.cu
sm80_gemm_f16_f16_f32_tensor_op_f32.cu
sm80_gemm_tf32_tf32_f32_tensor_op_f32.cu
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_sm90
BATCH_SOURCES ON
BATCH_SIZE 4
sm90_gemm_f16_f16_f16_tensor_op.cu
sm90_gemm_bf16_bf16_bf16_tensor_op_f32.cu
sm90_gemm_s8_s8_s8_tensor_op_s32.cu
sm90_gemm_tf32_tf32_f32_tensor_op_f32.cu
sm90_gemm_f32_f32_f32_tensor_op_f32.cu
)
# Alignment tests
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_alignx_sm90
BATCH_SOURCES ON
BATCH_SIZE 4
sm90_gemm_f16_f16_f16_alignx_tensor_op.cu
sm90_gemm_bf16_bf16_bf16_alignx_tensor_op_f32.cu
sm90_gemm_s8_s8_s8_alignx_tensor_op_s32.cu
sm90_gemm_tf32_tf32_f32_alignx_tensor_op_f32.cu
)
CUTLASS 3.1 (#915) Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2023-04-15 11:19:34 +08:00
# Fused epilogue tests
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_epilogue_fusion_sm90
BATCH_SOURCES ON
BATCH_SIZE 4
sm90_gemm_f16_f16_f16_tensor_op_f32_tensor_broadcast.cu
sm90_gemm_f32_f32_f32_tensor_op_f32_tensor_broadcast.cu
sm90_gemm_s8_s8_s8_tensor_op_s32_tensor_broadcast.cu
sm90_gemm_f16_f16_f16_tensor_op_f32_cluster_warpspecialized_cooperative_bias_elementwise.cu
sm90_gemm_f16_f16_f16_tensor_op_f32_cluster_warpspecialized_pingpong_bias_elementwise.cu
)
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_cluster_multicast_sm90
BATCH_SOURCES ON
BATCH_SIZE 4
sm90_gemm_f16_f16_f16_tensor_op_f32_cluster_unspecialized.cu
sm90_gemm_f16_f16_f16_tensor_op_f32_cluster_warpspecialized.cu
CUTLASS 3.1 (#915) Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2023-04-15 11:19:34 +08:00
sm90_gemm_f16_f16_f16_tensor_op_f32_cluster_warpspecialized_pingpong.cu
sm90_gemm_f16_f16_f16_tensor_op_f32_cluster_warpspecialized_cooperative.cu
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_gmma_rs_warpspecialized_sm90
BATCH_SOURCES ON
BATCH_SIZE 4
sm90_gemm_tf32_tf32_f32_tensor_op_f32_gmma_rs_cluster_warpspecialized.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
)
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_f32_tf32_sm80
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_tf32t_tf32n_f32t_tensor_op_f32_sm80.cu
gemm_tf32n_tf32t_f32t_tensor_op_f32_sm80.cu
gemm_tf32n_tf32n_f32t_tensor_op_f32_sm80.cu
gemm_tf32t_tf32t_f32t_tensor_op_f32_sm80.cu
gemm_universal_cf32n_cf32n_cf32n_tensor_op_f32_sm80.cu
gemm_cf32n_cf32t_cf32t_tensor_op_tf32_f32_sm80.cu
gemm_cf32t_cf32n_cf32t_tensor_op_tf32_f32_sm80.cu
gemm_f32n_f32n_f32t_tensor_op_f32_sm80.cu
gemm_f32n_f32n_f32t_tensor_op_bf16_f32_sm80.cu
CUTLASS 2.6 (#298) CUTLASS 2.6
2021-07-23 12:40:53 +08:00
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
sm80_gemm_f16_f16_f32_tensor_op_f32.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_f64
BATCH_SOURCES ON
BATCH_SIZE 4
gemm_f64n_f64t_f64t_tensor_op_f64_sm80.cu
gemm_f64t_f64n_f64t_tensor_op_f64_sm80.cu
gemm_universal_cf64n_cf64t_cf64t_tensor_op_f64_sm80.cu
gemm_universal_cf64n_cf64t_cf64t_tensor_op_f64_gaussian_sm80.cu
gemm_cf64n_cf64t_cf64t_tensor_op_f64_sm80.cu
gemm_cf64t_cf64n_cf64t_tensor_op_f64_sm80.cu
gemm_cf64n_cf64t_cf64t_tensor_op_f64_gaussian_sm80.cu
gemm_cf64t_cf64n_cf64t_tensor_op_f64_gaussian_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# SM90 device level tests
gemm_f64n_f64t_f64t_tensor_op_f64_sm90.cu
gemm_f64t_f64n_f64t_tensor_op_f64_sm90.cu
CUTLASS 3.0.0 (#786) * CUTLASS 3.0.0
2023-01-24 09:55:28 +08:00
sm80_gemm_f64_f64_f64_tensor_op_f64.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
gemm_cf64n_cf64t_cf64t_tensor_op_f64_sm90.cu
gemm_cf64t_cf64n_cf64t_tensor_op_f64_sm90.cu
gemm_cf64n_cf64t_cf64t_tensor_op_f64_gaussian_sm90.cu
gemm_cf64t_cf64n_cf64t_tensor_op_f64_gaussian_sm90.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_s32_sm80
BATCH_SOURCES ON
BATCH_SIZE 4
gemm_s8t_s8n_s32t_tensor_op_s32_sm80.cu
gemm_s8t_s8n_s32n_tensor_op_s32_sm80.cu
gemm_s8t_s8n_s8n_tensor_op_s32_sm80.cu
gemm_s8t_s8n_s8t_tensor_op_s32_sm80.cu
CUTLASS 3.1 (#915) Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2023-04-15 11:19:34 +08:00
gemm_s8t_s8n_f16t_tensor_op_s32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_s4t_s4n_s32n_tensor_op_s32_sm80.cu
gemm_s4t_s4n_s32t_tensor_op_s32_sm80.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
gemm_s4t_s4n_s4n_tensor_op_s32_sm80.cu
gemm_s4t_s4n_s4t_tensor_op_s32_sm80.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_b1t_b1n_s32n_tensor_op_s32_sm80.cu
gemm_b1t_b1n_s32t_tensor_op_s32_sm80.cu
gemm_s8n_s8t_s8n_tensor_op_s32_sm80.cu
gemm_s4n_s4t_s4n_tensor_op_s32_sm80.cu
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_wmma
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
# wmma floating point tests
gemm_f16t_f16n_f16t_wmma_tensor_op_f16_sm70.cu
gemm_f16n_f16t_f16t_wmma_tensor_op_f16_sm70.cu
gemm_f16t_f16t_f16t_wmma_tensor_op_f16_sm70.cu
gemm_f16n_f16n_f16t_wmma_tensor_op_f16_sm70.cu
gemm_f16t_f16n_f16n_wmma_tensor_op_f16_sm70.cu
gemm_f16n_f16t_f16n_wmma_tensor_op_f16_sm70.cu
gemm_f16t_f16t_f16n_wmma_tensor_op_f16_sm70.cu
gemm_f16n_f16n_f16n_wmma_tensor_op_f16_sm70.cu
gemm_f16t_f16n_f32t_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16t_f32t_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16t_f32t_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16n_f32t_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16n_f32n_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16t_f32n_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16t_f32n_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16n_f32n_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16n_f16t_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16t_f16t_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16t_f16t_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16n_f16t_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16n_f16n_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16t_f16n_wmma_tensor_op_f32_sm70.cu
gemm_f16t_f16t_f16n_wmma_tensor_op_f32_sm70.cu
gemm_f16n_f16n_f16n_wmma_tensor_op_f32_sm70.cu
# wmma int8 tests
gemm_s8t_s8n_s32t_wmma_tensor_op_s32_sm72.cu
gemm_s8t_s8n_s32n_wmma_tensor_op_s32_sm72.cu
gemm_s8t_s8n_s8t_wmma_tensor_op_s32_sm72.cu
gemm_s8t_s8n_s8n_wmma_tensor_op_s32_sm72.cu
# wmma uint8 tests
gemm_u8t_u8n_s32t_wmma_tensor_op_s32_sm72.cu
# wmma sub byptes (s4 and b1) tests
gemm_s4t_s4n_s32n_wmma_tensor_op_s32_sm75.cu
gemm_s4t_s4n_s32t_wmma_tensor_op_s32_sm75.cu
gemm_b1t_b1n_s32n_wmma_tensor_op_s32_sm75.cu
gemm_b1t_b1n_s32t_wmma_tensor_op_s32_sm75.cu
# wmma floating point tests (using singestage pipeline)
gemm_f16t_f16n_f16t_singlestage_wmma_tensor_op_f16_sm70.cu
gemm_f16t_f16n_f16n_singlestage_wmma_tensor_op_f16_sm70.cu
gemm_f16t_f16n_f32t_singlestage_wmma_tensor_op_f32_sm70.cu
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_tensorop_planar_complex
BATCH_SOURCES ON
BATCH_SIZE 4
gemm_planar_complex_f16_f16_f32_tensor_op_sm70.cu
gemm_planar_complex_f16_f16_f32_tensor_op_sm75.cu
gemm_planar_complex_f16_f16_f32_tensor_op_sm80.cu
)
CUTLASS 2.8 (#363) CUTLASS 2.8
2021-11-20 05:26:35 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_grouped
BATCH_SOURCES ON
BATCH_SIZE 4
gemm_grouped_sm80.cu
)
CUTLASS 2.10 (#615) Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2022-09-04 06:48:46 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_grouped_scheduler
BATCH_SOURCES ON
BATCH_SIZE 4
gemm_grouped_scheduler_sm80.cu
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_grouped_rank_2k_scheduler
BATCH_SOURCES ON
BATCH_SIZE 4
rank_2k_grouped_scheduler_sm80.cu
)
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_sparse_tensorop_sm80
BATCH_SOURCES ON
BATCH_SIZE 4
CUTLASS 2.2 (#96) Adds support for NVIDIA Ampere Architecture features. CUDA 11 Toolkit recommended.
2020-06-09 07:17:35 +08:00
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
gemm_f16n_f16n_f16t_tensor_op_f32_sparse_sm80.cu
gemm_f16n_f16n_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f16n_f16t_f16t_tensor_op_f16_sparse_sm80.cu
gemm_f16n_f16t_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f16t_f16n_f16t_tensor_op_f16_sparse_sm80.cu
gemm_f16t_f16n_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f16t_f16t_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f32t_f32n_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f32n_f32t_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f32t_f32t_f32t_tensor_op_f32_sparse_sm80.cu
gemm_f32n_f32n_f32t_tensor_op_f32_sparse_sm80.cu
gemm_s8t_s8n_s32t_tensor_op_s32_sparse_sm80.cu
gemm_s4t_s4n_s32t_tensor_op_s32_sparse_sm80.cu
CUTLASS 2.0 (#62) CUTLASS 2.0 Substantially refactored for - Better performance, particularly for native Turing Tensor Cores - Robust and durable templates spanning the design space - Encapsulated functionality embodying modern C++11 programming techniques - Optimized containers and data types for efficient, generic, portable device code Updates to: - Quick start guide - Documentation - Utilities - CUTLASS Profiler Native Turing Tensor Cores - Efficient GEMM kernels targeting Turing Tensor Cores - Mixed-precision floating point, 8-bit integer, 4-bit integer, and binarized operands Coverage of existing CUTLASS functionality: - GEMM kernels targeting CUDA and Tensor Cores in NVIDIA GPUs - Volta Tensor Cores through native mma.sync and through WMMA API - Optimizations such as parallel reductions, threadblock rasterization, and intra-threadblock reductions - Batched GEMM operations - Complex-valued GEMMs Note: this commit and all that follow require a host compiler supporting C++11 or greater.
2019-11-20 08:55:34 +08:00
)
CUTLASS 2.4 (Implicit GEMM convolution) (#147) CUTLASS 2.4 (Implicit GEMM Convolution) Co-authored-by: Manish Gupta <manigupta@nvidia.com>, Haicheng Wu <haichengw@nvidia.com>, Dustyn Blasig <dblasig@nvidia.com>, Andrew Kerr <akerr@nvidia.com>
2020-11-20 13:25:25 +08:00
CUTLASS 2.6 (#298) CUTLASS 2.6
2021-07-23 12:40:53 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemv_device
BATCH_SOURCES ON
BATCH_SIZE 4
gemv.cu
)
if (NOT CUDA_COMPILER MATCHES "[Cc]lang")
add_dependencies(
cutlass_test_unit_gemm_device
cutlass_test_unit_gemm_device_gemm_with_fused_epilogue_tensorop
)
add_dependencies(
test_unit_gemm_device
test_unit_gemm_device_gemm_with_fused_epilogue_tensorop
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_gemm_with_fused_epilogue_tensorop
gemm_with_reduction_f16n_f16n_f16n_tensorop_f32_sm75.cu
gemm_with_broadcast_f16n_f16n_f16n_tensorop_f32_sm75.cu
gemm_with_reduction_f16t_f16n_f16n_tensorop_f32_sm80.cu
)
endif()
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
if (NOT CUDA_COMPILER MATCHES "[Cc]lang")
add_dependencies(
cutlass_test_unit_gemm_device
cutlass_test_unit_gemm_device_blas3
)
add_dependencies(
test_unit_gemm_device
test_unit_gemm_device_blas3
)
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_blas3
BATCH_SOURCES ON
BATCH_SIZE 4
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
## SYRK
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
# Syrk SM80 f64 tests
syrk_f64n_f64t_tensor_op_f64_sm80.cu
syrk_f64t_f64n_tensor_op_f64_sm80.cu
# Syrk SM80 f32 tests
syrk_tf32n_f32t_tensor_op_f32_sm80.cu
syrk_tf32t_f32t_tensor_op_f32_sm80.cu
syrk_f32n_f32t_tensor_op_fast_f32_sm80.cu
syrk_f32t_f32t_tensor_op_fast_f32_sm80.cu
# Syrk SM80 complex f64 tests
syrk_cf64n_cf64t_tensor_op_f64_sm80.cu
syrk_cf64n_cf64n_tensor_op_f64_sm80.cu
syrk_cf64n_cf64t_tensor_op_f64_gaussian_sm80.cu
# Syrk SM80 complex f32 tests
syrk_cf32n_cf32t_tensor_op_f32_sm80.cu
syrk_cf32n_cf32n_tensor_op_f32_sm80.cu
syrk_cf32n_cf32t_tensor_op_fast_f32_sm80.cu
syrk_cf32n_cf32n_tensor_op_fast_f32_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Syrk SM90 f64 tests
syrk_f64_f64_tensor_op_f64_sm90.cu
# Syrk SM90 complex f64 tests
syrk_cf64_cf64_tensor_op_f64_sm90.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
## HERK
# Herk SM80 complex f64 tests
herk_cf64h_cf64n_tensor_op_f64_sm80.cu
# Herk SM80 complex f32 tests
herk_cf32h_cf32n_tensor_op_f32_sm80.cu
herk_cf32h_cf32n_tensor_op_fast_f32_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Herk SM90 complex f64 tests
herk_cf64_cf64_tensor_op_f64_sm90.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
## TRMM
# Trmm SM80 f64 tests
trmm_f64n_f64n_f64t_tensor_op_f64_ls_sm80.cu
trmm_f64n_f64n_f64t_tensor_op_f64_rs_sm80.cu
trmm_f64t_f64t_f64n_tensor_op_f64_ls_sm80.cu
trmm_f64t_f64t_f64n_tensor_op_f64_rs_sm80.cu
trmm_f64n_f64t_f64t_tensor_op_f64_rs_sm80.cu
# Trmm SM80 f32 tests
trmm_tf32t_tf32n_f32t_tensor_op_f32_ls_sm80.cu
trmm_tf32n_tf32t_f32t_tensor_op_f32_ls_sm80.cu
trmm_tf32n_tf32t_f32t_tensor_op_f32_rs_sm80.cu
trmm_tf32t_tf32n_f32n_tensor_op_f32_ls_sm80.cu
trmm_f32t_f32n_f32t_tensor_op_fast_f32_ls_sm80.cu
trmm_f32n_f32t_f32t_tensor_op_fast_f32_ls_sm80.cu
trmm_f32n_f32t_f32t_tensor_op_fast_f32_rs_sm80.cu
trmm_f32t_f32n_f32n_tensor_op_fast_f32_ls_sm80.cu
# Trmm SM80 complex f64 tests
trmm_cf64n_cf64n_cf64t_tensor_op_f64_sm80.cu
trmm_cf64n_cf64n_cf64t_tensor_op_f64_gaussian_sm80.cu
# Trmm SM80 complex f32 tests
trmm_cf32n_cf32n_cf32t_tensor_op_f32_sm80.cu
trmm_cf32n_cf32n_cf32t_tensor_op_fast_f32_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Trmm SM90 f64 tests
trmm_f64_f64_f64_tensor_op_f64_sm90.cu
# Trmm SM90 complex f64 tests
trmm_cf64_cf64_cf64_tensor_op_f64_sm90.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
## SYR2K
# Syr2k SM80 f64 tests
syr2k_f64n_f64t_tensor_op_f64_sm80.cu
syr2k_f64n_f64n_tensor_op_f64_sm80.cu
syr2k_f64t_f64n_tensor_op_f64_sm80.cu
# Syr2k SM80 f32 tests
syr2k_tf32n_f32n_tensor_op_f32_sm80.cu
syr2k_tf32t_f32n_tensor_op_f32_sm80.cu
syr2k_f32n_f32n_tensor_op_fast_f32_sm80.cu
syr2k_f32t_f32n_tensor_op_fast_f32_sm80.cu
# Syr2k SM80 complex f64 tests
syr2k_cf64n_cf64t_tensor_op_f64_sm80.cu
syr2k_cf64n_cf64n_tensor_op_f64_sm80.cu
# Syr2k SM80 complex f32 tests
syr2k_cf32n_cf32n_tensor_op_f32_sm80.cu
syr2k_cf32n_cf32t_tensor_op_f32_sm80.cu
syr2k_cf32n_cf32n_tensor_op_fast_f32_sm80.cu
syr2k_cf32n_cf32t_tensor_op_fast_f32_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Syr2k SM90 f64 tests
syr2k_f64_f64_tensor_op_f64_sm90.cu
# Syr2k SM90 complex f64 tests
syr2k_cf64_cf64_tensor_op_f64_sm90.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
## HER2K
# Her2k SM80 complex f64 tests
her2k_cf64n_cf64n_tensor_op_f64_sm80.cu
# Her2k SM80 complex f32 tests
her2k_cf32h_cf32n_tensor_op_f32_sm80.cu
her2k_cf32h_cf32n_tensor_op_fast_f32_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Her2k SM90 complex f64 tests
her2k_cf64_cf64_tensor_op_f64_sm90.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
## SYMM
# Symm SM80 f64 tests
symm_f64n_f64n_tensor_op_f64_ls_sm80.cu
symm_f64n_f64n_tensor_op_f64_rs_sm80.cu
symm_f64n_f64t_tensor_op_f64_ls_sm80.cu
symm_f64n_f64t_tensor_op_f64_rs_sm80.cu
symm_f64t_f64n_tensor_op_f64_ls_sm80.cu
symm_f64t_f64n_tensor_op_f64_rs_sm80.cu
symm_f64t_f64t_tensor_op_f64_ls_sm80.cu
symm_f64t_f64t_tensor_op_f64_rs_sm80.cu
# Symm SM80 f32 tests
symm_tf32n_f32n_tensor_op_f32_ls_sm80.cu
symm_tf32n_f32n_tensor_op_f32_rs_sm80.cu
symm_tf32t_f32t_tensor_op_f32_ls_sm80.cu
symm_f32n_f32n_tensor_op_fast_f32_ls_sm80.cu
symm_f32n_f32n_tensor_op_fast_f32_rs_sm80.cu
symm_f32t_f32t_tensor_op_fast_f32_ls_sm80.cu
# Symm SM80 complex f64 tests
symm_cf64n_cf64n_cf64n_tensor_op_ls_f64_sm80.cu
symm_cf64n_cf64n_cf64n_tensor_op_rs_f64_sm80.cu
symm_cf64n_cf64n_cf64n_tensor_op_ls_f64_gaussian_sm80.cu
# Symm SM80 complex f32 tests
symm_cf32n_cf32n_tensor_op_f32_ls_sm80.cu
symm_cf32n_cf32n_tensor_op_f32_rs_sm80.cu
symm_cf32n_cf32n_tensor_op_fast_f32_ls_sm80.cu
symm_cf32n_cf32n_tensor_op_fast_f32_rs_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Symm SM90 f64 tests
symm_f64_f64_tensor_op_f64_sm90.cu
# Symm SM90 complex f64 tests
symm_cf64_cf64_cf64_tensor_op_f64_sm90.cu
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
# Hemm SM80 complex f64 tests
hemm_cf64h_cf64n_cf64n_tensor_op_ls_f64_sm80.cu
hemm_cf64h_cf64n_cf64n_tensor_op_rs_f64_sm80.cu
hemm_cf64h_cf64n_cf64n_tensor_op_ls_f64_gaussian_sm80.cu
# Hemm SM80 complex f32 tests
hemm_cf32h_cf32n_tensor_op_f32_ls_sm80.cu
hemm_cf32h_cf32n_tensor_op_f32_rs_sm80.cu
hemm_cf32h_cf32n_tensor_op_fast_f32_ls_sm80.cu
hemm_cf32h_cf32n_tensor_op_fast_f32_rs_sm80.cu
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
# Hemm SM90 complex f64 tests
hemm_cf64_cf64_cf64_tensor_op_f64_sm90.cu
)
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
CUTLASS 2.10 (#615) Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2022-09-04 06:48:46 +08:00
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_grouped_blas3
BATCH_SOURCES ON
BATCH_SIZE 4
# Grouped SYR2K SM80 f64 tests
syr2k_f64n_f64n_tensor_op_f64_grouped_sm80.cu
syr2k_f64n_f64t_tensor_op_f64_grouped_sm80.cu
syr2k_f64t_f64n_tensor_op_f64_grouped_sm80.cu
syr2k_f64t_f64t_tensor_op_f64_grouped_sm80.cu
# Grouped SYR2K SM80 cf64 tests
syr2k_cf64n_cf64n_tensor_op_f64_grouped_sm80.cu
syr2k_cf64n_cf64t_tensor_op_f64_grouped_sm80.cu
syr2k_cf64t_cf64n_tensor_op_f64_grouped_sm80.cu
syr2k_cf64t_cf64t_tensor_op_f64_grouped_sm80.cu
# Grouped HER2K SM80 f64 tests
her2k_cf64n_cf64n_tensor_op_f64_grouped_sm80.cu
her2k_cf64h_cf64n_tensor_op_f64_grouped_sm80.cu
)
CUTLASS 2.9 (#468)
2022-04-24 03:02:38 +08:00
endif()
releaase 2.11 (#703)
2022-11-19 22:02:15 +08:00
if (NOT CUDA_COMPILER MATCHES "[Cc]lang")
cutlass_test_unit_add_executable(
cutlass_test_unit_gemm_device_broadcast
gemm_f16t_f16n_f16t_tensor_op_f16_broadcast_sm80.cu
)
endif()
Reference in New Issue Copy Permalink