fix bug in DPCPPGuardImpl::getStream (#1701)

mingxiaoh · web-flow · commit a9c26467fc7e · 2022-10-25T17:33:32.000+08:00
diff --git a/csrc/aten/core/GuardImpl.cpp b/csrc/aten/core/GuardImpl.cpp
@@ -41,7 +41,7 @@ void DPCPPGuardImpl::uncheckedSetDevice(Device d) const noexcept {
 }
 
 Stream DPCPPGuardImpl::getStream(Device d) const noexcept {
-  return getCurrentDPCPPStream().unwrap();
+  return getCurrentDPCPPStream(d.index()).unwrap();
 }
 
 Stream DPCPPGuardImpl::exchangeStream(Stream s) const noexcept {

Original file line number	Diff line number	Diff line change
`@@ -41,7 +41,7 @@ void DPCPPGuardImpl::uncheckedSetDevice(Device d) const noexcept {`
`41`	`41`	`}`
`42`	`42`
`43`	`43`	`Stream DPCPPGuardImpl::getStream(Device d) const noexcept {`
`44`		`- return getCurrentDPCPPStream().unwrap();`
	`44`	`+ return getCurrentDPCPPStream(d.index()).unwrap();`
`45`	`45`	`}`
`46`	`46`
`47`	`47`	`Stream DPCPPGuardImpl::exchangeStream(Stream s) const noexcept {`