Initial commit.

Author: maleadt

LICENSE.md

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright © 2018 Tim Besard, and other contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

README.md

@@ -0,0 +1,10 @@
+NCCL.jl
+=======
+
+*A Julia wrapper for the NVIDIA Collective Communications Library.*
+
+
+Status
+------
+
+This package is very incomplete, and completely untested.

REQUIRE

@@ -0,0 +1 @@
+CUDAapi 0.4.0

deps/.gitignore

@@ -0,0 +1,2 @@
+ext.jl
+ext.jl.bak

deps/build.jl

@@ -0,0 +1,65 @@
+using CUDAapi
+
+const config_path = joinpath(@__DIR__, "ext.jl")
+const previous_config_path = config_path * ".bak"
+
+function write_ext(config, path)
+    open(path, "w") do io
+        println(io, "# autogenerated file, do not edit")
+        for (key,val) in config
+            println(io, "const $key = $(repr(val))")
+        end
+    end
+end
+
+function read_ext(path)
+    config = Dict{Symbol,Any}()
+    r = r"^const (\w+) = (.+)$"
+    open(path, "r") do io
+        for line in eachline(io)
+            m = match(r, line)
+            if m != nothing
+                config[Symbol(m.captures[1])] = eval(Meta.parse(m.captures[2]))
+            end
+        end
+    end
+    return config
+end
+
+function main()
+    ispath(config_path) && mv(config_path, previous_config_path; force=true)
+    config = Dict{Symbol,Any}(:configured => false)
+    write_ext(config, config_path)
+
+
+    ## discover stuff
+
+    toolkit_dirs = CUDAapi.find_toolkit()
+
+    config[:libnccl] = CUDAapi.find_cuda_library("nccl", toolkit_dirs)
+    if config[:libnccl] == nothing
+        error("could not find NCCL")
+    end
+
+    config[:configured] = true
+
+
+    ## (re)generate ext.jl
+
+    if isfile(previous_config_path)
+        @debug("Checking validity of existing ext.jl...")
+        previous_config = read_ext(previous_config_path)
+
+        if config == previous_config
+            @info "NCCL.jl has already been built for this toolchain, no need to rebuild"
+            mv(previous_config_path, config_path; force=true)
+            return
+        end
+    end
+
+    write_ext(config, config_path)
+
+    return
+end
+
+main()

src/NCCL.jl

@@ -0,0 +1,13 @@
+module NCCL
+
+const ext = joinpath(dirname(@__DIR__), "deps", "ext.jl")
+isfile(ext) || error("NCCL.jl has not been built, please run Pkg.build(\"NCCL\").")
+include(ext)
+if !configured
+    # default (non-functional) values for critical variables,
+    # making it possible to _load_ the package at all times.
+    const libnccl = nothing
+end
+
+
+end