namespace contrib {
using namespace runtime;
-
+namespace {
inline CBLAS_TRANSPOSE BooleanToTranspose(bool trans) { return trans ? CblasTrans : CblasNoTrans; }
inline CBLAS_OFFSET StringToOffset(const std::string offset_type) {
}
inline char BooleanToTransposeChar(bool trans) { return trans ? 'T' : 'N'; }
+} // namespace
struct MKLGemmU8S8S32Op {
void operator()(bool ta, bool tb, int M, int N, int K, float alpha, const void* A, int lda,
namespace contrib {
using namespace runtime;
-
+namespace {
inline cublasOperation_t BooleanToTranspose(bool item) { return item ? CUBLAS_OP_T : CUBLAS_OP_N; }
inline void TryEnableTensorCore(cublasHandle_t hdl) {
CHECK_CUBLAS_ERROR(cublasGetVersion(hdl, &version));
if (version >= 9000) CHECK_CUBLAS_ERROR(cublasSetMathMode(hdl, CUBLAS_TENSOR_OP_MATH));
}
+} // namespace
struct CublasHgemmOp {
typedef half TDatatype;