Initial import: grid-bot — grid trading bot for BTC-USDT on Cifra Markets
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
#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 "Types.h"
|
||||
#if !defined(__aarch64__)
|
||||
|
||||
#include <cstdint>
|
||||
#include "./FbgemmBuild.h" // @manual
|
||||
#include "./UtilsAvx2.h" // @manual
|
||||
|
||||
/// @defgroup fbgemm-quant-utils-avx512 Quantization Utilities (AVX512)
|
||||
///
|
||||
|
||||
namespace fbgemm {
|
||||
|
||||
/// @ingroup fbgemm-quant-utils-avx512
|
||||
///
|
||||
/// Requantize with AVX512.
|
||||
template <
|
||||
bool A_SYMMETRIC,
|
||||
bool B_SYMMETRIC,
|
||||
QuantizationGranularity Q_GRAN,
|
||||
bool HAS_BIAS,
|
||||
bool FUSE_RELU,
|
||||
int C_PER_G,
|
||||
typename BIAS_TYPE = std::int32_t>
|
||||
FBGEMM_API void requantizeOutputProcessingGConvAvx512(
|
||||
std::uint8_t* out,
|
||||
const std::int32_t* inp,
|
||||
const block_type_t& block,
|
||||
int ld_out,
|
||||
int ld_in,
|
||||
const requantizationParams_t<BIAS_TYPE>& r);
|
||||
|
||||
template <bool scale_bias_last = true, bool quant_padding_float_type = true>
|
||||
void Fused8BitRowwiseQuantizedSBFloatToBfloat16Avx512(
|
||||
const std::uint8_t* input,
|
||||
size_t input_rows,
|
||||
int input_columns,
|
||||
bfloat16* output);
|
||||
} // namespace fbgemm
|
||||
|
||||
#endif
|
||||
|
||||
#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)
|
||||
Reference in New Issue
Block a user