37 lines
835 B
C++
37 lines
835 B
C++
#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
|
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
|
|
#include "fbgemm/Utils.h"
|
|
|
|
namespace fbgemm {
|
|
|
|
FBGEMM_API void cblas_gemm_i64_i64acc(
|
|
matrix_op_t transa,
|
|
matrix_op_t transb,
|
|
int M,
|
|
int N,
|
|
int K,
|
|
const std::int64_t* A,
|
|
int lda,
|
|
const std::int64_t* B,
|
|
int ldb,
|
|
bool accumulate,
|
|
std::int64_t* C,
|
|
int ldc);
|
|
|
|
} // namespace fbgemm
|
|
|
|
#else
|
|
#error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
|
|
#endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|