Skip to content

Commit 36dee09

Browse files
shino16crcrpar
andcommitted
Add dynamic shape handling to nvfuser.full
Co-authored-by: crcrpar <[email protected]>
1 parent 948fe47 commit 36dee09

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

thunder/executors/nvfuserex_impl.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1045,10 +1045,11 @@ def full(
10451045
) -> Any:
10461046
nv_fill_value = getnv(fill_value, fd, lc_to_nv_map)
10471047
nvdtype = lcdtype_to_nvdtype(dtype)
1048+
nv_shape = getnv(shape, fd, lc_to_nv_map)
10481049

10491050
_select_device(fd, device)
10501051

1051-
return fd.ops.full(shape, nv_fill_value, nvdtype)
1052+
return fd.ops.full(nv_shape, nv_fill_value, nvdtype)
10521053

10531054

10541055
register_supported(PrimIDs.FULL, full, _full_check)

0 commit comments

Comments
 (0)