Skip to content
Snippets Groups Projects

Fix fc impl

Merged Maxence Naud requested to merge fix_fc_impl into dev
1 file
+ 2
2
Compare changes
  • Side-by-side
  • Inline
+ 2
2
@@ -30,7 +30,7 @@ void Aidge::FCImpl_cuda::forward() {
const auto& fcOp = static_cast<const FC_Op&>(mOp);
bool noBias = fcOp.template getAttr<FCAttr::NoBias>();
std::size_t outChannels = static_cast<std::size_t>(fcOp.template getAttr<FCAttr::OutChannels>());
std::size_t outChannels = fcOp.outChannels();
const auto& input0 = fcOp.getInput(0)->refCastFrom(mInput0Fallback, *fcOp.getOutput(0));
const auto& input1 = fcOp.getInput(1)->refCastFrom(mInput1Fallback, *fcOp.getOutput(0));
@@ -114,4 +114,4 @@ void Aidge::FCImpl_cuda::forward_(const Tensor& input0, const Tensor& input1, co
cudaFree(onesVector);
}
}
\ No newline at end of file
}
Loading