| 16 | |
| 17 | namespace { |
| 18 | void run_tensor_add( |
| 19 | Handle* handle_rocm, const TensorND& a, const TensorND& b, const TensorND& c) { |
| 20 | auto opr = handle_rocm->create_operator<ElemwiseForward>(); |
| 21 | opr->param().mode = ElemwiseForward::Mode::ADD; |
| 22 | hipProfilerStart(); |
| 23 | opr->exec({a, b}, c); |
| 24 | hipProfilerStop(); |
| 25 | } |
| 26 | |
| 27 | using Mode = ElemwiseForward::Mode; |
| 28 | template <Mode mode> |