Skip to content

Commit e099dbb

Browse files
committed
removing device kwargs since not used
1 parent 7ce66ea commit e099dbb

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

py/torch_tensorrt/dynamo/conversion/ops_evaluators.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,6 @@ def aten_ops_rand(
7373
kwargs: Dict[str, Argument],
7474
name: str,
7575
) -> Union[TRTTensor, Sequence[TRTTensor]]:
76-
device = kwargs.get("device", None)
7776
return np.random.rand(*args)
7877

7978

@@ -87,7 +86,6 @@ def aten_ops_randn(
8786
kwargs: Dict[str, Argument],
8887
name: str,
8988
) -> Union[TRTTensor, Sequence[TRTTensor]]:
90-
device = kwargs.get("device", None)
9189
return np.random.randn(*args)
9290

9391

@@ -119,5 +117,4 @@ def aten_ops_randperm(
119117
kwargs: Dict[str, Argument],
120118
name: str,
121119
) -> Union[TRTTensor, Sequence[TRTTensor]]:
122-
device = kwargs.get("device", None)
123120
return np.random.permutation(*args)

0 commit comments

Comments
 (0)