Pytorch-Correlation-extension icon indicating copy to clipboard operation
Pytorch-Correlation-extension copied to clipboard

Published 20 hours ago • verified publisher icon ClementPinard

torch.compile fullgraph=True

Open bhack opened this issue 2 years ago • 6 comments
trafficstars

It seems that this section it is not compilable fullgraph:

torch._dynamo.exc.Unsupported: call_method UserDefinedObjectVariable(forward) __call__ [TensorVariable(), TensorVariable(), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int), ConstantVariable(int)] {}

from user code:
   File "/opt/conda/lib/python3.10/site-packages/spatial_correlation_sampler/spatial_correlation_sampler.py", line 105, in forward
    return SpatialCorrelationSamplerFunction.apply(input1, input2, self.kernel_size,
  File "/opt/conda/lib/python3.10/site-packages/spatial_correlation_sampler/spatial_correlation_sampler.py", line 66, in forward
    output = correlation.forward(input1, input2,

bhack avatar Jun 17 '23 17:06 bhack