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 a7c541c commit a0d8604Copy full SHA for a0d8604
tripy/nvtripy/frontend/dimension_size.py
@@ -50,17 +50,15 @@ def eval(self) -> "nvtripy.DimensionSize":
50
"""
51
Immediately evaluates this ``DimensionSize`` object.
52
53
- .. note:: ``DimensionSize`` will always reside on host even it is evaluated.
+ .. note:: ``DimensionSize`` will always reside on host even after it is evaluated.
54
55
Returns:
56
The evaluated ``DimensionSize``.
57
58
.. code-block:: python
59
:linenos:
60
61
- import time
62
63
- start = time.perf_counter()
64
dim_size = tp.ones((2, 2)).shape[0]
65
dim_size.eval()
66
print(dim_size.device)
0 commit comments