ONE icon indicating copy to clipboard operation
ONE copied to clipboard

Run While Op with HE(tvn+cpu) via single circle model

Open chunseoklee opened this issue 2 years ago • 4 comments

Why

  1. Let's consider a single circle model which contains a operation like TRIX instead of partitioned model.

Originally posted by @chunseoklee in https://github.com/Samsung/ONE/issues/9300#issuecomment-1162600017

All the tasks for https://github.com/Samsung/ONE/projects/9 assume that we have multiple models in nnpacakge. However, inserting TRIX(or Bulk)-like Op into single circle model could be another solution.

What

  • Let's depict the whole workflow in this approach
  • Let's list up todo items to support this

chunseoklee avatar Jul 05 '22 06:07 chunseoklee

cc @ragmani

chunseoklee avatar Jul 05 '22 06:07 chunseoklee

Let's depict the whole workflow in this approach

Let's imagine input and output circle file of this workflow.

Assume that our input circle(A.circle) is like:

Conv2D(op1) - Conv2D(op2) - FC(op3) - Softmax(op4)

Also, if we'd like to run op1,op2 in TRIX backend and to run others in CPU backend

Then, our output circle is like:

Bulk(or CustomOp) - FC - Softmax

Of course, we have trix binary for Bulk layer in nnpackage as well as circle.

chunseoklee avatar Jul 05 '22 07:07 chunseoklee

@chunseoklee Will you prepare A.circle? Do you have any existing graph which contains this pattern? Or do we need to create it manually?

Conv2D(op1) - Conv2D(op2) - FC(op3) - Softmax(op4)

glistening avatar Jul 06 '22 01:07 glistening

@chunseoklee Will you prepare A.circle? Do you have any existing graph which contains this pattern? Or do we need to create it manually?

Conv2D(op1) - Conv2D(op2) - FC(op3) - Softmax(op4)

No. It is just a example to explain.

chunseoklee avatar Jul 06 '22 02:07 chunseoklee