Skip to content

Commit

Permalink
update compact and unplug CUDA
Browse files Browse the repository at this point in the history
  • Loading branch information
yuehhua committed Sep 6, 2023
1 parent cba3bb0 commit 6f0ca0a
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 18 deletions.
22 changes: 11 additions & 11 deletions Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@ authors = ["Yueh-Hua Tu <[email protected]>"]
version = "0.13.10"

[deps]
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
DataStructures = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8"
DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab"
Expand All @@ -15,36 +14,37 @@ Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
MLDatasets = "eb30cadb-4394-5ae3-aed4-317e484a6458"
NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"
Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
Requires = "ae029012-a4dd-5104-9daa-d747884805df"
SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
Word2Vec = "c64b6f0f-98cd-51d1-af78-58ae84944834"

[compat]
CUDA = "3, 4"
ChainRulesCore = "1"
DataStructures = "0.18"
FillArrays = "0.13, 1"
Flux = "0.12 - 0.13"
GraphSignals = "0.7 - 0.8"
DelimitedFiles = "1"
FillArrays = "1"
Flux = "0.14"
GraphSignals = "0.9"
Graphs = "1"
MLDatasets = "0.7"
NNlib = "0.8"
NNlibCUDA = "0.2"
Optimisers = "0.2"
NNlib = "0.9"
Optimisers = "0.2, 0.3"
Reexport = "1"
Requires = "1"
StatsBase = "0.33, 0.34"
Word2Vec = "0.5"
julia = "1.6"
julia = "1.9"

[extras]
Clustering = "aaaa29a8-35af-508c-8bc3-b662a17a0fe5"
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

[targets]
test = ["Clustering", "SparseArrays", "Test"]
test = ["Clustering", "CUDA", "SparseArrays", "Test"]
3 changes: 1 addition & 2 deletions src/GeometricFlux.jl
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,14 @@ using LinearAlgebra
using Random
using Reexport

using CUDA, CUDA.CUSPARSE
using ChainRulesCore
using ChainRulesCore: @non_differentiable
using FillArrays: Fill
using Flux
using Flux: glorot_uniform, leakyrelu, GRUCell, @functor
@reexport using GraphSignals
using Graphs
using NNlib, NNlibCUDA
using NNlib
using Optimisers

import Word2Vec: word2vec, wordvectors, get_vector
Expand Down
11 changes: 6 additions & 5 deletions test/runtests.jl
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
using LinearAlgebra
using SparseArrays: SparseMatrixCSC
using Statistics: mean
using Test

using GeometricFlux
using GeometricFlux.Datasets
using Clustering
Expand All @@ -7,11 +12,7 @@ using Flux: @functor
using FillArrays
using GraphSignals
using Graphs
using LinearAlgebra
using NNlib, NNlibCUDA
using SparseArrays: SparseMatrixCSC
using Statistics: mean
using Test
using NNlib

cuda_tests = [
"cuda/graph_conv",
Expand Down

0 comments on commit 6f0ca0a

Please sign in to comment.