Hi all,
I was wondering how can I call a specific llvm intrinsic from within a compute node.
Something on the line of:
def test(x):
def _compute(*indices):
value = x(*indices)
return call_llvm_aarch64_intrinsic(value)
return te.compute(x.shape, _compute)
The intrinsic I want to use would be applied as an element-wise operation (similar to pow
, log
, etc…)
This post from @anijain2305 is similar to what I would like to do, but got no answers: Broadcast operation for a vector
Is there any way to achieve what I am asking without going through tensorize
?
Thanks,
Giuseppe