diff --git a/docker/Dockerfile.ci_gpu_rocm b/docker/Dockerfile.ci_gpu_rocm index d8758f41a464..3f700c3f456e 100644 --- a/docker/Dockerfile.ci_gpu_rocm +++ b/docker/Dockerfile.ci_gpu_rocm @@ -13,7 +13,7 @@ ENV MAX_JOBS=${ARG_MAX_JOBS:-24} # Install basic tools RUN apt-get update \ - && apt install -y clang-format-15 ninja-build tree + && apt install -y clang-format-15 git ninja-build tree #Updating to latest version of Cmake for Ubuntu 22.04 RUN apt-get update && apt-get install -y wget build-essential \ diff --git a/python/dgl/graphbolt/impl/ondisk_dataset.py b/python/dgl/graphbolt/impl/ondisk_dataset.py index cf1c144ceb70..f51bc1f84fed 100644 --- a/python/dgl/graphbolt/impl/ondisk_dataset.py +++ b/python/dgl/graphbolt/impl/ondisk_dataset.py @@ -136,8 +136,8 @@ def _graph_data_to_fused_csc_sampling_graph( src, dst = read_edges(dataset_dir, edge_fmt, edge_path) edge_type_offset.append(edge_type_offset[-1] + len(src)) src_type, _, dst_type = etype_str_to_tuple(edge_info["type"]) - src += node_type_offset[node_type_to_id[src_type]] - dst += node_type_offset[node_type_to_id[dst_type]] + src = src + node_type_offset[node_type_to_id[src_type]] + dst = dst + node_type_offset[node_type_to_id[dst_type]] coo_src_list.append(torch.tensor(src)) coo_dst_list.append(torch.tensor(dst)) coo_etype_list.append(torch.full((len(src),), etype_id)) diff --git a/tests/python/pytorch/test_ffi-stream.py b/tests/python/pytorch/test_ffi-stream.py index 4700a5d3ebf1..7853a4eba7e9 100644 --- a/tests/python/pytorch/test_ffi-stream.py +++ b/tests/python/pytorch/test_ffi-stream.py @@ -160,6 +160,13 @@ def perform_computing(): @unittest.skipIf( F._default_context_str == "cpu", reason="stream only runs on GPU." ) +@unittest.skipIf( + F.is_hip(), + reason=( + "ROCm allocator reuse differs from CUDA; this negative test depends on " + "nondeterministic block reuse." + ), +) def test_record_stream_graph_negative(): cycles_per_ms = _get_cycles_per_ms() diff --git a/tests/scripts/task_unit_test_rocm.sh b/tests/scripts/task_unit_test_rocm.sh old mode 100644 new mode 100755