-
Notifications
You must be signed in to change notification settings - Fork 874
[API Compatibility] Add some device related API #7467
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
|
感谢你贡献飞桨文档,文档预览构建中,Docs-New 跑完后即可预览,预览链接:http://preview-pr-7467.paddle-docs-preview.paddlepaddle.org.cn/documentation/docs/zh/api/index_cn.html |
📚 本次 PR 文档预览链接(点击展开)
|
| current_stream | ||
|
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| current_stream | |
| current_stream | |
| ------------------------------- |
| • None:获取当前设备上的默认计算流。 | ||
| • int:设备索引,例如 0 表示 cuda:0。 | ||
| • str:设备字符串,例如 'cuda:0' 或 'gpu:1'。 | ||
| • CUDAPlace:Paddle 的 CUDAPlace 对象。 | ||
| • CustomPlace:Paddle 的自定义设备 Place 对象。 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
参数不是只有 device 吗?其他的如果是参数的说明,就再加一层缩进。另外,不要用 • 用 -
|
|
||
| 参数 | ||
| ::::::::: | ||
| • device (int | str | CUDAPlace | CustomPlace | None, optional) – 指定需要同步的设备。 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
同样的问题,参数不是只有 device 吗?其他的如果是参数的说明,就再加一层缩进。另外,不要用 • 用 -
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从``get_rng_state()``获取 | ||
| - **device** (DeviceLike, 可选) - 要设置 RNG 状态的设备: | ||
|
|
||
| - 如果不指定,则使用当前默认设备(由``paddle.framework._current_expected_place_()``返回) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| - 如果不指定,则使用当前默认设备(由``paddle.framework._current_expected_place_()``返回) | |
| - 如果不指定,则使用当前默认设备(由 ``paddle.framework._current_expected_place_()`` 返回) |
|
|
||
| 参数 | ||
| ::::::::: | ||
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从``get_rng_state()``获取 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从``get_rng_state()``获取 | |
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从 ``get_rng_state()`` 获取 |
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从``get_rng_state()``获取 | ||
| - **device** (DeviceLike, 可选) - 要设置 RNG 状态的设备: | ||
|
|
||
| - 如果不指定,则使用当前默认设备(由``paddle.framework._current_expected_place_()``返回) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| - 如果不指定,则使用当前默认设备(由``paddle.framework._current_expected_place_()``返回) | |
| - 如果不指定,则使用当前默认设备(由 ``paddle.framework._current_expected_place_()`` 返回) |
|
|
||
| 参数 | ||
| ::::::::: | ||
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从``get_rng_state()``获取 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从``get_rng_state()``获取 | |
| - **new_state** (core.GeneratorState) - 要设置的 RNG 状态对象,通常从 ``get_rng_state()`` 获取 |
|
|
||
| 异常 | ||
| ::::::::: | ||
| - RuntimeError: 如果设备类型是 CPU(Paddle 不暴露`paddle.cpu`模块)或找不到匹配的设备模块 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| - RuntimeError: 如果设备类型是 CPU(Paddle 不暴露`paddle.cpu`模块)或找不到匹配的设备模块 | |
| - RuntimeError: 如果设备类型是 CPU(Paddle 不暴露 ``paddle.cpu`` 模块)或找不到匹配的设备模块 |
Co-authored-by: zachary sun <[email protected]>
sunzhongkai588
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
" :ref:
paddle.get_default_device <cn_api_paddle_get_default_device>", "获取当前默认设备信息"" :ref:
paddle.get_device_module <cn_api_paddle_get_device_module>", "获取指定设备对应的模块"" :ref:
current_stream <cn_api_paddle_cuda_current_stream>", "获取当前 CUDA 流"" :ref:
get_device_properties <cn_api_paddle_cuda_get_device_properties>", "获取 CUDA 设备属性"" :ref:
get_rng_state <cn_api_paddle_cuda_get_rng_state>", "获取随机数生成器状态"" :ref:
is_available <cn_api_paddle_cuda_is_available>", "检查 CUDA 是否可用"" :ref:
is_current_stream_capturing <cn_api_paddle_cuda_is_current_stream_capturing>", "检查当前流是否处于捕获状态"" :ref:
is_current_stream_capturing <cn_api_paddle_device_is_current_stream_capturing>", "检查当前流是否处于捕获状态"