Simple multiplication function.
(self, x, y)
| 40 | |
| 41 | @I.pyfunc |
| 42 | def multiply(self, x, y): |
| 43 | """Simple multiplication function.""" |
| 44 | x_tvm = self._convert_pytorch_to_tvm(x) |
| 45 | y_tvm = self._convert_pytorch_to_tvm(y) |
| 46 | result = self.call_tir( |
| 47 | self.multiply_tir, [x_tvm, y_tvm], out_sinfo=R.Tensor((5,), "float32") |
| 48 | ) |
| 49 | return self._convert_tvm_to_pytorch(result) |
| 50 | |
| 51 | @T.prim_func(s_tir=True) |
| 52 | def add_tir(var_x: T.handle, var_y: T.handle, var_out: T.handle): |
nothing calls this directly
no test coverage detected