From ed86137cac6815cfe95b6e6e8938447094111c02 Mon Sep 17 00:00:00 2001 From: Anton Reinhard Date: Wed, 8 May 2024 19:01:32 +0200 Subject: [PATCH] Enable oneAPI and ROCm --- Project.toml | 4 +++- src/MetagraphOptimization.jl | 6 ++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/Project.toml b/Project.toml index c495944..63f9d2e 100644 --- a/Project.toml +++ b/Project.toml @@ -1,9 +1,10 @@ -authors = ["Anton Reinhard "] name = "MetagraphOptimization" uuid = "3e869610-d48d-4942-ba70-c1b702a33ca4" +authors = ["Anton Reinhard "] version = "0.1.0" [deps] +AMDGPU = "21141c5a-9bdb-4563-92ae-f87d6854732e" AccurateArithmetic = "22286c92-06ac-501d-9306-4abd417d9753" CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa" @@ -18,6 +19,7 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c" Roots = "f2b01f46-fcfa-551c-844a-d8ac1e96c665" StaticArrays = "90137ffa-7385-5640-81b9-e52037218182" UUIDs = "cf7118a7-6976-5b1a-9a39-7adc72f591a4" +oneAPI = "8f75cd03-7ff8-4ecb-9b8f-daf728133b1b" [extras] CUDA_Runtime_jll = "76a88914-d11a-5bdc-97e0-2f5a05c973a2" diff --git a/src/MetagraphOptimization.jl b/src/MetagraphOptimization.jl index f17c8b1..608b515 100644 --- a/src/MetagraphOptimization.jl +++ b/src/MetagraphOptimization.jl @@ -196,10 +196,8 @@ include("devices/impl.jl") include("devices/numa/impl.jl") include("devices/cuda/impl.jl") -# can currently not use AMDGPU because of incompatability with the newest rocm drivers -# include("devices/rocm/impl.jl") -# oneapi seems also broken for now -# include("devices/oneapi/impl.jl") +include("devices/rocm/impl.jl") +include("devices/oneapi/impl.jl") include("scheduler/interface.jl") include("scheduler/greedy.jl") -- 2.49.0