AI/rocm-systems
2
0
Derivar 0
Código Questões Pedidos de integração Operações Pacotes Planeamentos Lançamentos Wiki Trabalho

Fix/RO Backend Hang Issue (#53)

Ver a proveniência 
* Update HIP version check for compatibility with versions >= 5.5

* Update memory allocator for context BlockHandle
   - Replaced `HIPAllocator` with `HIPDefaultFinegrainedAllocator` for context `BlockHandle`.

* Update run commands for `rocshmem_g` and `rocshmem_p` functional tests

[ROCm/rocshmem commit: c16b0d6952]
Este cometimento está contido em:
Avinash Kethineedi
2025-03-24 22:54:07 -05:00
cometido por GitHub
ascendente 1ee9b72449
cometimento baca5fd7a1
3 ficheiros modificados com 12 adições e 15 eliminações
Descarregar ficheiro patch Descarregar ficheiro diff Expandir todos os ficheiros Colapsar todos os ficheiros
projects/rocshmem/scripts/functional_tests/driver.sh
+8 -12
Ver ficheiro
@@ -181,19 +181,15 @@ TestRMA() {
ExecTest "teamctxget" 2 4 128 1024
ExecTest "teamctxget" 2 16 256 1024
ExecTest "g" 2 1 1 1048576
ExecTest "g" 2 1 1024 512
ExecTest "g" 2 8 1 1048576
ExecTest "g" 2 16 128 8
ExecTest "g" 2 32 256 512
ExecTest "g" 2 64 1024 8
ExecTest "g" 2 1 1 128
ExecTest "g" 2 1 1024 2
ExecTest "g" 2 8 1 32
ExecTest "g" 2 16 128 4
ExecTest "p" 2 1 1 1048576
ExecTest "p" 2 1 1024 512
ExecTest "p" 2 8 1 1048576
ExecTest "p" 2 16 128 8
ExecTest "p" 2 32 256 512
ExecTest "p" 2 64 1024 8
ExecTest "p" 2 1 1 128
ExecTest "p" 2 1 1024 2
ExecTest "p" 2 8 1 32
ExecTest "p" 2 16 128 4
################################ Non-Blocking ################################
projects/rocshmem/src/memory/hip_allocator.hpp
+2 -1
Ver ficheiro
@@ -37,7 +37,8 @@
#include "memory_allocator.hpp"
// `hipDeviceMallocUncached` was introduced at ROCm 5.5
#if HIP_VERSION_MAJOR >= 5 && HIP_VERSION_MINOR >= 5
#if (HIP_VERSION_MAJOR > 5) || \
(HIP_VERSION_MAJOR == 5 && HIP_VERSION_MINOR >= 5)
#define HIP_SUPPORTS_MALLOC_UNCACHED
#endif
namespace rocshmem {
projects/rocshmem/src/reverse_offload/block_handle.hpp
+2 -2
Ver ficheiro
@@ -83,7 +83,7 @@ class DefaultBlockHandleProxy {
ProxyT proxy_{};
};
using DefaultBlockHandleProxyT = DefaultBlockHandleProxy<HIPAllocator>;
using DefaultBlockHandleProxyT = DefaultBlockHandleProxy<HIPDefaultFinegrainedAllocator>;
template <typename ALLOCATOR>
class BlockHandleProxy {
@@ -130,7 +130,7 @@ class BlockHandleProxy {
size_t num_blocks_{};
};
using BlockHandleProxyT = BlockHandleProxy<HIPAllocator>;
using BlockHandleProxyT = BlockHandleProxy<HIPDefaultFinegrainedAllocator>;
} // namespace rocshmem