diff options
Diffstat (limited to 'compiler/nnc/tests/soft_backend/test_main.def')
| -rw-r--r-- | compiler/nnc/tests/soft_backend/test_main.def | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/compiler/nnc/tests/soft_backend/test_main.def b/compiler/nnc/tests/soft_backend/test_main.def index 6a464f862..c508cca26 100644 --- a/compiler/nnc/tests/soft_backend/test_main.def +++ b/compiler/nnc/tests/soft_backend/test_main.def @@ -1,8 +1,9 @@ +#include <string> int main() { Shape s{1, 2, 3}; Tensor in_t(s); - NNModel model("nnmodel.params"); + NNModel model(std::string("nnmodel.params")); model.set_in(in_t); model.doInference(); std::shared_ptr<Tensor> out_t = model.get_out(); |