Entire Frame [?]: 1.580207586288452
Backend [?]: 0.21279644966125488
Inductor [?]: 0.1510319709777832
Code Gen Time: 0.0851740837097168
No failures!
Restart Reasons:
Graph break due to unsupported builtin spatial_correlation_sampler_backend.PyCapsule.forward. This function is either a Python builtin (e.g. _warnings.warn) or a third-party C/C++ Python extension (perhaps created with pybind). If it is a Python builtin, please file an issue on GitHub so the PyTorch team can add support for it and see the next case for a workaround. If it is a third-party C/C++ Python extension, please either wrap it into a PyTorch-understood custom operator (see https://pytorch.org/docs/main/notes/custom_operators.html for more details) or, if it is traceable, use torch.compiler.allow_in_graph.
Cache Size: 2
Accumulated Cache Size: 2
Guard Count: 92
Shape Env Guards: 12
Graph Ops: 11
Graph Nodes: 17
Graph Inputs: 5
Compliant Custom Ops:
Non-Compliant Custom Ops:
Sym | Source(s) | Value | User stack | Framework stack |
---|---|---|---|---|
s0 | L['v'].size()[2] |
68 |
|
|
s1 | L['v'].size()[3] |
68 |
|
|
s2 | L['q'].size()[2] |
68 |
|
|
s3 | L['q'].size()[3] |
68 |
|
|
s4 | L['k'].size()[2] |
68 |
|
|
s5 | L['k'].size()[3] |
68 |
|
|