Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 1a5fd078 authored by hhakim's avatar hhakim
Browse files

Add MatBSR<FPP, GPU2> getNbBlockRow/Col and getNBlocks functions + update to gpu_mod@1488445b.

parent 9edb74f7
No related branches found
No related tags found
No related merge requests found
Subproject commit b5966c7445654ab15d3ce87c0f289cb4bbf93248 Subproject commit 1488445be70f76d3ca0db37ee7703bffbfefdec1
...@@ -76,6 +76,28 @@ namespace Faust ...@@ -76,6 +76,28 @@ namespace Faust
return bsr_funcs->get_nbytes(gpu_mat); return bsr_funcs->get_nbytes(gpu_mat);
} }
template<>
size_t MatBSR<@FAUST_SCALAR_FOR_GM@, GPU2>::getNbBlockRow() const
{
auto bsr_funcs = GPUModHandler::get_singleton()->bsr_funcs((@FAUST_SCALAR_FOR_GM@)(0));
return bsr_funcs->get_bnrows(gpu_mat);
}
template<>
size_t MatBSR<@FAUST_SCALAR_FOR_GM@, GPU2>::getNbBlockCol() const
{
auto bsr_funcs = GPUModHandler::get_singleton()->bsr_funcs((@FAUST_SCALAR_FOR_GM@)(0));
return bsr_funcs->get_bncols(gpu_mat);
}
template<>
size_t MatBSR<@FAUST_SCALAR_FOR_GM@, GPU2>::getNBlocks() const
{
auto bsr_funcs = GPUModHandler::get_singleton()->bsr_funcs((@FAUST_SCALAR_FOR_GM@)(0));
return bsr_funcs->get_bnnz(gpu_mat);
}
template<> template<>
int32_t MatBSR<@FAUST_SCALAR_FOR_GM@, GPU2>::getDevice() const int32_t MatBSR<@FAUST_SCALAR_FOR_GM@, GPU2>::getDevice() const
{ {
......
...@@ -51,7 +51,10 @@ namespace Faust ...@@ -51,7 +51,10 @@ namespace Faust
/*********** MatGeneric member functions **************/ /*********** MatGeneric member functions **************/
void setZeros(); void setZeros();
size_t getNbBlockRow() const;
size_t getNbBlockCol() const;
size_t getNBytes() const; size_t getNBytes() const;
size_t getNBlocks() const;
MatType getType() const; MatType getType() const;
int32_t getNbRow() const; int32_t getNbRow() const;
int32_t getNbCol() const; int32_t getNbCol() const;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment