Skip to content

Device Graph Runner增加异步语义 #486

Description

@LeoLau94

Describe the feature🚀

现在Device Graph Runner只有同步语义,会阻塞当前的线程和stream,我们希望有一套兼容异步语义回调和runner forward的设计

assert self._copy_to_param_buffer(*args, **kwargs)

References

https://docs.pytorch.org/docs/stable/generated/torch.cuda.Stream.html
https://docs.pytorch.org/docs/main/notes/cuda.html

Environment details

XpuGrpah: 0.10.1

Metadata

Metadata

Labels

enhancementNew feature or request
No fields configured for Feature.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions