When compiling models containing torch_scatter.scatter_max operations with torch.compile, Dynamo fails during FX graph tracing due to a fake tensor allocation error ...
[RANK 0] ### funcol ### [RANK 0] AsyncCollectiveTensor(tensor([[[ 0, 4, 8], [12, 16, 20]]], device='cuda:0')) [RANK 0] ### golden ### [RANK 0] tensor([[[ 0, 8, 16 ...
Abstract: The development of a prototype Compton scattering camera based on two 1 cm/sup 3/ cubic 3-D position sensitive CZT detectors is underway in our laboratory ...