Skip to content

请问如何设置CUSTOM_DEVICE_BLACK_LIST ，让nn.SimpleRNN算子执行在CPU上，其余跑在加速器上？ #1548

New issue
New issue
Open
Open
请问如何设置CUSTOM_DEVICE_BLACK_LIST ，让nn.SimpleRNN算子执行在CPU上，其余跑在加速器上？#1548
Assignees
YanhuiDua
@wangzy0327

Description

@wangzy0327
wangzy0327
opened on Jan 8, 2025

image
我这里脚本设置后，所有算子操作均跑在CPU上了

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions