Skip to content
Snippets Groups Projects
Commit a9f688e3 authored by Maxence Naud's avatar Maxence Naud
Browse files

Merge branch 'fix_fc_impl' into 'dev'

Fix fc impl

See merge request !23
parents 8cef5883 cc60a931
No related branches found
No related tags found
2 merge requests!32version 0.2.1,!23Fix fc impl
Pipeline #47203 passed
...@@ -30,7 +30,7 @@ void Aidge::FCImpl_cuda::forward() { ...@@ -30,7 +30,7 @@ void Aidge::FCImpl_cuda::forward() {
const auto& fcOp = static_cast<const FC_Op&>(mOp); const auto& fcOp = static_cast<const FC_Op&>(mOp);
bool noBias = fcOp.template getAttr<FCAttr::NoBias>(); 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& input0 = fcOp.getInput(0)->refCastFrom(mInput0Fallback, *fcOp.getOutput(0));
const auto& input1 = fcOp.getInput(1)->refCastFrom(mInput1Fallback, *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 ...@@ -114,4 +114,4 @@ void Aidge::FCImpl_cuda::forward_(const Tensor& input0, const Tensor& input1, co
cudaFree(onesVector); cudaFree(onesVector);
} }
} }
\ No newline at end of file
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