Updated opmult to non-template
This commit is contained in:
parent
cd1ee97ce8
commit
2a45f5f6af
1 changed files with 1 additions and 1 deletions
|
|
@ -103,7 +103,7 @@ int main() {
|
|||
auto tensorOut = mgr.buildTensor({ 0., 0., 0. });
|
||||
|
||||
// 3. Run multiplication operation synchronously
|
||||
mgr.evalOpDefault<kp::OpMult<>>(
|
||||
mgr.evalOpDefault<kp::OpMult>(
|
||||
{ tensorInA, tensorInB, tensorOut })
|
||||
|
||||
// 4. Map results back from GPU memory to print the results
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue