Skip to content

Commit

Permalink
Correcting typographical errors, and removing useless imports.
Browse files Browse the repository at this point in the history
  • Loading branch information
codetalker7 committed Jul 25, 2023
1 parent f8fc22c commit 3cd1d89
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 4 deletions.
4 changes: 2 additions & 2 deletions ext/FluxAMDGPUExt/FluxAMDGPUExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@ const MIOPENFloat = AMDGPU.MIOpen.MIOPENFloat
# Set to boolean on the first call to check_use_amdgpu
const USE_AMDGPU = Ref{Union{Nothing, Bool}}(nothing)

Flux.isavailable(device::Flux.FluxAMDGPUDevice) = true
Flux.isfunctional(device::Flux.FluxAMDGPUDevice) = AMDGPU.functional()
Flux.isavailable(device::Flux.FluxAMDDevice) = true
Flux.isfunctional(device::Flux.FluxAMDDevice) = AMDGPU.functional()

function check_use_amdgpu()
if !isnothing(USE_AMDGPU[])
Expand Down
2 changes: 1 addition & 1 deletion ext/FluxMetalExt/FluxMetalExt.jl
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ using Zygote
const USE_METAL = Ref{Union{Nothing, Bool}}(nothing)

Flux.isavailable(device::Flux.FluxMetalDevice) = true
Flux.isfunctional(devuce::Flux.FluxMetalDevice) = Metal.functional()
Flux.isfunctional(device::Flux.FluxMetalDevice) = Metal.functional()

function check_use_metal()
isnothing(USE_METAL[]) || return
Expand Down
1 change: 0 additions & 1 deletion src/Flux.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@ using Preferences
using LinearAlgebra, Statistics, Random # standard lib
using MacroTools, Reexport, ProgressLogging, SpecialFunctions
using MacroTools: @forward
import Base: PkgId, UUID

@reexport using NNlib
using MLUtils
Expand Down

0 comments on commit 3cd1d89

Please sign in to comment.