Merge pull request #120 from JuliaGPU/jps/load-everywhere

Allow AMDGPU to load everywhere

Author: jpsamaroo

.buildkite/pipeline.yml

@@ -14,6 +14,8 @@ steps:
       rocm: "*"
     if: build.message !~ /\[skip tests\]/
     timeout_in_minutes: 60
+    env:
+      JULIA_AMDGPU_HSA_MUST_LOAD: "1"
 
   - label: "Documentation"
     plugins:

.ci/load.jl

@@ -0,0 +1,12 @@
+import Pkg
+
+Pkg.update()
+
+root_directory = dirname(@__DIR__)
+
+amdgpu = Pkg.PackageSpec(path = root_directory)
+Pkg.develop(amdgpu)
+Pkg.build()
+Pkg.precompile()
+
+import AMDGPU

.github/workflows/ci-julia-1.6-nightly.yml

@@ -0,0 +1,46 @@
+name: CI (Julia 1.6-nightly)
+on:
+  push:
+    branches:
+      - master
+    tags: '*'
+  pull_request:
+defaults:
+  run:
+    shell: bash
+jobs:
+  CI-julia-1-6-nightly:
+    name: CI-julia-1-6-nightly
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        version:
+          - '1.6-nightly'
+        os:
+          - ubuntu-latest
+          - macOS-latest
+          - windows-latest
+        arch:
+          - x64
+    steps:
+      - uses: actions/checkout@v2
+      - uses: julia-actions/setup-julia@v1
+        with:
+          version: ${{ matrix.version }}
+          arch: ${{ matrix.arch }}
+      - uses: actions/cache@v1
+        env:
+          cache-name: cache-artifacts
+        with:
+          path: ~/.julia/artifacts
+          key: ${{ runner.os }}-test-${{ env.cache-name }}-${{ hashFiles('**/Project.toml') }}
+          restore-keys: |
+            ${{ runner.os }}-test-${{ env.cache-name }}-
+            ${{ runner.os }}-test-
+            ${{ runner.os }}-
+      - run: julia .ci/load.jl
+      - uses: julia-actions/julia-processcoverage@v1
+      - uses: codecov/codecov-action@v1
+        with:
+          file: lcov.info

Project.toml

@@ -1,7 +1,7 @@
 name = "AMDGPU"
 uuid = "21141c5a-9bdb-4563-92ae-f87d6854732e"
 authors = ["Julian P Samaroo <jpsamaroo@jpsamaroo.me>"]
-version = "0.2.4"
+version = "0.2.5"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"

deps/build.jl

@@ -258,7 +258,7 @@ if dl_info === nothing && unsatisfied
     # Alternatively, you could attempt to install from a separate provider,
     # build from source or something even more ambitious here.
     @warn "Your platform (\"$(Sys.MACHINE)\", parsed as \"$(triplet(platform_key_abi()))\") is not supported by this package!"
-    return
+    exit(0)
 end
 
 # If we have a download, and we are unsatisfied (or the version we're

src/AMDGPU.jl

@@ -156,6 +156,16 @@ function __init__()
         HSA runtime has not been built, runtime functionality will be unavailable.
         Please run Pkg.build("AMDGPU") and reload AMDGPU.
         """
+
+        if parse(Bool, get(ENV, "JULIA_AMDGPU_HSA_MUST_LOAD", "0"))
+            println("Diagnostics:")
+            println("-- deps/build.log")
+            println(String(read(joinpath(@__DIR__, "..", "deps", "build.log"))))
+            println("-- permissions")
+            run(`ls -lah /dev/kfd`)
+            run(`ls -lah /dev/dri`)
+            run(`id`)
+        end
     end
 
     # Check whether device intrinsics are available