GB4996

Graph-Break Type

Short name describing what triggered the graph break

torch.cuda.use_mem_pool with non-constant device

Context

Values or code snippet captured at the break point

device={device}

Explanation

Explanation of why the graph break was triggered

Dynamo requires the device argument to torch.cuda.use_mem_pool to be a Python constant.

Hints

Hints on how to resolve the graph break

No hints provided.

Additional Information

Click here to add Additional Info

Back to Registry