We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 948fe47 commit 36dee09Copy full SHA for 36dee09
thunder/executors/nvfuserex_impl.py
@@ -1045,10 +1045,11 @@ def full(
1045
) -> Any:
1046
nv_fill_value = getnv(fill_value, fd, lc_to_nv_map)
1047
nvdtype = lcdtype_to_nvdtype(dtype)
1048
+ nv_shape = getnv(shape, fd, lc_to_nv_map)
1049
1050
_select_device(fd, device)
1051
- return fd.ops.full(shape, nv_fill_value, nvdtype)
1052
+ return fd.ops.full(nv_shape, nv_fill_value, nvdtype)
1053
1054
1055
register_supported(PrimIDs.FULL, full, _full_check)
0 commit comments