Skip to content

Commit 9b594d9

Browse files
Export c10::hip::HIPCachingAllocatorMasqueradingAsCUDA::get() in libtorch_hip.so so apex sees symbol definition at runtime
1 parent 97a8a89 commit 9b594d9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

aten/src/ATen/hip/impl/HIPCachingAllocatorMasqueradingAsCUDA.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ class DataPtr;
1010
namespace hip {
1111
namespace HIPCachingAllocatorMasqueradingAsCUDA {
1212

13-
Allocator* get();
13+
C10_HIP_API Allocator* get();
1414
C10_HIP_API void recordStreamMasqueradingAsCUDA(const DataPtr& ptr, HIPStreamMasqueradingAsCUDA stream);
1515

1616
} // namespace HIPCachingAllocatorMasqueradingAsCUDA

0 commit comments

Comments
 (0)