removing device kwargs since not used

pytorch · Apr 4, 2024 · e099dbb · e099dbb
1 parent 7ce66ea
commit e099dbb
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/py/torch_tensorrt/dynamo/conversion/ops_evaluators.py b/py/torch_tensorrt/dynamo/conversion/ops_evaluators.py
@@ -73,7 +73,6 @@ def aten_ops_rand(
     kwargs: Dict[str, Argument],
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    device = kwargs.get("device", None)
     return np.random.rand(*args)
 
 
@@ -87,7 +86,6 @@ def aten_ops_randn(
     kwargs: Dict[str, Argument],
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    device = kwargs.get("device", None)
     return np.random.randn(*args)
 
 
@@ -119,5 +117,4 @@ def aten_ops_randperm(
     kwargs: Dict[str, Argument],
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    device = kwargs.get("device", None)
     return np.random.permutation(*args)