About tvm_address_of


#1

When I look at Use Tensorize to Leverage Hardware doc, and print out s, the function tvm_address_of exists in the loop, as shown below

produce C {
for (i, 0, 1024) {
for (j.outer, 0, 32) {
gemv_update(tvm_address_of(C[(((i32) + j.outer)16)]), tvm_address_of(A[(i64)]), tvm_address_of(B[(j.outer1024)]), 16, 64, 64)
}
}
}

What does tvm_address_of do?


#2

It takes the address of the corresponding element and pass it as an argument.


#3

thank you for your help,Then,I have a question about use tensorize ,I want to use a special architectural language for tensorization, but I find that tensorize parameters can only be tensors, what should I do to achieve my goal


#4

It would be great if you can create a separate thread with code examples to describe what you need. atm it is hard to tell what is your requirement