Author: Jinen Setpal
Tags: Distance
Description: A direct replacement of torch's cdist, including support for backprop. Doesn't take torch's compute_mode argument.
Minimal Usage:
import triton_cdist # registers operator
# from here you can use it as a stand-in replacement of `torch.cdist`.
x1 = ...
x2 = ...
p = ...
torch.ops.triton_cdist.opt_cdist(x1, x2, p=p) # previously, `torch.cdist(x1, x1, p=p)`Triton Version: v3.3.1+
Id in triton index: 0016