/* ************************************************************************ * Copyright 2019-2021 Advanced Micro Devices, Inc. * ************************************************************************ */ #pragma once #include "Tensile/gemm.hpp" #include "definitions.hpp" template ROCBLAS_KERNEL void set_matrix_zero_if_alpha_zero_kernel(rocblas_int m, rocblas_int n, TScal alpha_device_host, rocblas_stride stride_alpha, TPtr Aa, rocblas_int lda, rocblas_stride a_st_or_of); template rocblas_status set_matrix_zero_if_alpha_zero_template(rocblas_handle handle, rocblas_int m, rocblas_int n, TScal alpha, rocblas_stride stride_alpha, TPtr A, rocblas_int lda, rocblas_stride a_st_or_of, rocblas_int batch_count); template ROCBLAS_INTERNAL_EXPORT_NOINLINE rocblas_status rocblas_internal_trmm_outofplace_template(rocblas_handle handle, rocblas_side side, rocblas_fill uplo, rocblas_operation trans_a, rocblas_diagonal diag, rocblas_int m, rocblas_int n, TScal* alpha, rocblas_stride stride_alpha, TConstPtr* dA, rocblas_int offset_a, rocblas_int lda, rocblas_stride stride_a, TConstPtr* dB, rocblas_int offset_b, rocblas_int ldb, rocblas_stride stride_b, TPtr* dC, rocblas_int offset_c, rocblas_int lddc, rocblas_stride stride_c, rocblas_int batch_count); template ROCBLAS_INTERNAL_EXPORT_NOINLINE rocblas_status rocblas_internal_trmm_recursive_inplace_template(rocblas_handle handle, rocblas_side side, rocblas_fill uplo, rocblas_operation trans_a, rocblas_diagonal diag, rocblas_int m, rocblas_int n, TScal* alpha, rocblas_stride stride_alpha, TConstPtr* dA, rocblas_stride offset_a, rocblas_int lda, rocblas_stride stride_a, TPtr* dB, rocblas_stride offset_b, rocblas_int ldb, rocblas_stride stride_b, rocblas_int batch_count); template ROCBLAS_INTERNAL_EXPORT_NOINLINE rocblas_status rocblas_internal_trmm_template(rocblas_handle handle, rocblas_side side, rocblas_fill uplo, rocblas_operation trans_a, rocblas_diagonal diag, rocblas_int m, rocblas_int n, TScal* alpha, rocblas_stride stride_alpha, TConstPtr* dA, rocblas_int offset_a, rocblas_int lda, rocblas_stride stride_a, TConstPtr* dB, rocblas_int offset_b, rocblas_int ldb, rocblas_stride stride_b, TPtr* dC, rocblas_int offset_c, rocblas_int lddc, rocblas_stride stride_c, rocblas_int batch_count);