Skip to content

Commit 6004c65

Browse files
kevinstephanopytorchmergebot
authored andcommitted
Fix rand_like nvprim meta function. (pytorch#85882)
Really minor fix necessary to work with TorchDynamo. Pull Request resolved: pytorch#85882 Approved by: https://github.com/mruberry, https://github.com/jjsjann123
1 parent 103a21f commit 6004c65

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/_prims/nvfuser_prims.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -331,6 +331,7 @@ def _meta_rand_like(
331331
):
332332
strides = make_contiguous_strides_for(self.shape)
333333
return torch._prims.TensorMeta(
334+
self,
334335
shape=self.shape,
335336
strides=strides,
336337
dtype=dtype,

0 commit comments

Comments
 (0)