Skip to content

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

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

Fields

No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions