Skip to content

Commit 80d9a01

Browse files
committed
fix musaMalloc issue
1 parent c9eafb5 commit 80d9a01

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

flagcx/adaptor/musa_adaptor.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,7 @@ flagcxResult_t musaAdaptorDeviceMalloc(void **ptr, size_t size,
4646
if (type == flagcxMemHost) {
4747
DEVCHECK(musaMallocHost(ptr, size));
4848
} else if (type == flagcxMemDevice) {
49-
if (stream == NULL) {
50-
DEVCHECK(musaMalloc(ptr, size));
51-
} else {
52-
DEVCHECK(musaMallocAsync(ptr, size, stream->base));
53-
}
49+
DEVCHECK(musaMalloc(ptr, size));
5450
} else if (type == flagcxMemManaged) {
5551
DEVCHECK(musaMallocManaged(ptr, size, musaMemAttachGlobal));
5652
}

0 commit comments

Comments
 (0)