Skip to content
Snippets Groups Projects
Commit 6ab74383 authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

minor changes

parent 4a65163a
No related branches found
No related tags found
2 merge requests!54Update 0.3.1 -> 0.4.0,!45Add support for the MatMul operator
Pipeline #68123 passed
......@@ -174,7 +174,7 @@ void crossLayerEqualization(std::shared_ptr<GraphView> graphView, double targetD
insertScalingBelowProducer(n1->getParent(1), s1, graphView);
if (n1->type() != "MatMul") // TODO : enhance this !
if (n1->type() != "MatMul") // TODO : exclude every node that we can't call getParent(2) on !
if (n1->getParent(2))
insertScalingBelowProducer(n1->getParent(2), s1, graphView);
......
......@@ -56,8 +56,8 @@ void Aidge::LSQImpl_cuda::backward() {
if (mWorkspace != nullptr) {
cudaFree(mWorkspace);
}
std::size_t sizeofData = getDataTypeBitWidth(gra_int0->dataType()) / 8;
CHECK_CUDA_STATUS(cudaMalloc(&mWorkspace, sizeofData * gra_int0->size()));
std::size_t sizeOfData = getDataTypeBitWidth(gra_int0->dataType()) / 8;
CHECK_CUDA_STATUS(cudaMalloc(&mWorkspace, sizeOfData * gra_int0->size()));
mWorkspaceSize = gra_int0->size();
}
......
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