Hey, I am testing out various TVM pathways for doing general data processing
tasks.
One thing that seemed like low hanging fruit was kmeans++. I fairly quickly
hit an interesting snag that I am curious about.
I was going to attempt to do the diff-squared calculation in the reduction
itself and hit an error. In order to do this I attempted to declare a
commreducer that had 1 lhs argument (the summation) and 2 rhs arguments (the
two elemwise items of the two vectors):
```clojure
expanded-distances-op (ast/compute
[n-rows n-centers]
(ast/tvm-fn
[row-idx center-idx]
(ast/commutative-reduce
(ast/tvm-fn->commutative-reducer
(ast/tvm-fn
[sum row-elem center-elem]
(ast/tvm-let [diff (ast-op/- row-elem
center-elem)]
(ast-op/+ sum (ast-op/mul diff
diff))))
[(float 0.0)])
[{:domain [0 n-cols] :name "elem-idx"}]
[(fn [elem-idx]
(ast/tget dataset [row-idx elem-idx]))
(fn [elem-idx]
(ast/tget dataset [(ast/tget center-indexes
[center-idx])
elem-idx]))]))
"expanded-distances")
```
This hit an error in expr.cc that I was very curious about:
https://github.com/apache/incubator-tvm/blob/728b829575e5e690870b111ae2256cbe0f3dbe6f/src/tir/ir/expr.cc#L827
I can in this case and in general lift a compound statement out into an
intermediary but I thought the restriction that I had to have an equal number
of accumulators and inputs to a reduction was very odd.
Is perhaps this first check not necessary? Are there other work arounds (such
as creating a tuple) I could try beside simply creating another compute
operation?
Thanks for your time, I really like the work done on the object model and am
having a great time playing with TVM.
---
[Visit
Topic](https://discuss.tvm.apache.org/t/multiple-right-hand-sides-for-reduction/8457/1)
to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.tvm.apache.org/email/unsubscribe/6c74acd77f1bc2ad0f3e120c0ee8eb1688bc0c1d88a9222f4352800f745872bd).
