Skip to content
Snippets Groups Projects
Commit fd321fc0 authored by Jerome Hue's avatar Jerome Hue
Browse files

Fix forward signature of NFixedBitFlip op

parent ae873a18
No related branches found
No related tags found
No related merge requests found
......@@ -23,7 +23,7 @@ namespace Aidge {
template<typename T2, typename T1>
struct BitCastHelper {
static constexpr bool isValid =
static constexpr bool isValid =
sizeof(T1) == sizeof(T2) and
std::is_trivially_copyable<T1>::value and
std::is_trivially_copyable<T2>::value and
......@@ -61,7 +61,7 @@ template<typename T>
typename std::enable_if<(sizeof(T) != 4 and sizeof(T) != 8)>::type
flipSingleBit(T& vlaue, std::size_t bitToFlip)
{
AIDGE_THROW_OR_ABORT(std::runtime_error,
AIDGE_THROW_OR_ABORT(std::runtime_error,
"BitFlip error not implemented for this type of data.");
}
......@@ -69,11 +69,11 @@ template <class I, class O>
void FixedNBitFlipImpl_cpu_forward_kernel(const void* input_,
void* output_,
std::size_t inputLength,
std::uint32_t nBitToFlip,
std::vector<int> weights,
std::vector<int> bits) {
std::size_t nBitToFlip,
std::vector<std::size_t> weights,
std::vector<uint8_t> bits) {
const I* input = static_cast<const I*>(input_);
O* output = static_cast<O*>(output_);
......@@ -107,7 +107,7 @@ void FixedNBitFlipImpl_cpu_forward_kernel(const void* input_,
O oldValue = output[randomIndex];
flipSingleBit(output[randomIndex], bitToFlip);
O newValue = output[randomIndex];
Log::notice("\t Flipping bit {} of weight {}.\tFrom {} to {}",
bitToFlip,
randomIndex,
......@@ -121,20 +121,19 @@ void FixedNBitFlipImpl_cpu_forward_kernel(const void* input_,
// Kernels registration to implementation entry point
REGISTRAR(FixedNBitFlipImpl_cpu,
{DataType::Float32},
{ProdConso::inPlaceModel,
Aidge::FixedNBitFlipImpl_cpu_forward_kernel<float, float>,
{ProdConso::inPlaceModel,
Aidge::FixedNBitFlipImpl_cpu_forward_kernel<float, float>,
nullptr});
REGISTRAR(FixedNBitFlipImpl_cpu,
{DataType::Float64},
{ProdConso::inPlaceModel,
Aidge::FixedNBitFlipImpl_cpu_forward_kernel<double, double>,
{ProdConso::inPlaceModel,
Aidge::FixedNBitFlipImpl_cpu_forward_kernel<double, double>,
nullptr});
REGISTRAR(FixedNBitFlipImpl_cpu,
{DataType::Int32},
{ProdConso::inPlaceModel,
Aidge::FixedNBitFlipImpl_cpu_forward_kernel<std::uint32_t, std::uint32_t>,
{ProdConso::inPlaceModel,
Aidge::FixedNBitFlipImpl_cpu_forward_kernel<std::uint32_t, std::uint32_t>,
nullptr});
} // namespace Aidge
#endif
......@@ -4,8 +4,8 @@
namespace Aidge {
static constexpr const int PROJECT_VERSION_MAJOR = 0;
static constexpr const int PROJECT_VERSION_MINOR = 6;
static constexpr const int PROJECT_VERSION_PATCH = 0;
static constexpr const char * PROJECT_VERSION = "0.6.0";
static constexpr const char * PROJECT_GIT_HASH = "1af5777";
static constexpr const int PROJECT_VERSION_PATCH = 1;
static constexpr const char * PROJECT_VERSION = "0.6.1";
static constexpr const char * PROJECT_GIT_HASH = "ae873a1";
}
#endif // VERSION_H
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment