Hi， @tqchen
Now, i want to call one device api in the compute, For example:
output = tvm.extern((1,), [input], lambda ins, outs:
tvm.call_extern(“int32_t”, device_func_name,
n, ins[0].access_ptr(“r”), incx, outs[0].access_ptr(“w”)),
name=“output”, dtype = inp_dtype)
but, when the device api params include the struct, eg:
int32_t cc_device_exp(const Dtype scale, const Dtype shift, const Dtype base,
kTensorShape tensorShape, const Dtype * x, Dtype *y)
ps: the tensorShape is struct
How to describe the compute dsl?
In the backend of tvm, there is some info about the struct, but, not find the demo
constexpr const char* tvm_struct_set = “tvm_struct_set”
pls give help
Thinks