Updated opmult to non-template

This commit is contained in:
Alejandro Saucedo 2020-12-19 18:21:47 +00:00
parent cd1ee97ce8
commit 2a45f5f6af

View file

@ -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