search

microsoft / nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
MIT License
948 stars 158 forks source link

[Fix] nnfusion jit inference fix #448

Closed LeiWang1999 closed 2 years ago

LeiWang1999 commented 2 years ago
  1. currently nnfusion jit do not support float16 input datatype, fix it. refers to issue 446
  2. the function get_workspace_size return an big value, but the return type is int and it will throw a overflow exception, fix it. refers to issue 447