Skip to content

Commit

Permalink
[DTensor] Added op_call in no-mesh dispatch assert message (pytorch…
Browse files Browse the repository at this point in the history
…#113903)

This helps debug, e.g. when there is an unsupported op.
Pull Request resolved: pytorch#113903
Approved by: https://github.com/wanchaol
ghstack dependencies: pytorch#113654
  • Loading branch information
awgu authored and pytorchmergebot committed Nov 17, 2023
1 parent 0894981 commit 99b89db
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/distributed/_tensor/dispatch.py
Original file line number Diff line number Diff line change
Expand Up @@ -326,7 +326,7 @@ def unwrap_to_op_info(
kwargs_schema[k] = v
local_kwargs[k] = v

assert mesh is not None, "found no DeviceMesh from dtensor args!"
assert mesh is not None, f"found no DeviceMesh from dtensor args for {op_call}!"
op_info = OpInfo(
mesh,
OpSchema(
Expand Down

0 comments on commit 99b89db

Please sign in to comment.