Skip to content

Commit 69b9950

Browse files
author
Avik Pal
committed
travis test on julia1.0
1 parent 2495c1b commit 69b9950

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

.travis.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ os:
44
- linux
55
- osx
66
julia:
7-
- 0.7
7+
- 1.0
88
notifications:
99
email: false
1010
git:

src/nnpack/interface.jl

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,10 @@ function check_support(x, k, pad, stride, dilation = 1)
66
return pad_, stride_, fallback
77
end
88

9-
#NOTE: Commenting out the activation functions until sure what to do
10-
11-
# relu(x::AA1) = nnp_relu_output(x, inplace ? x : similar(x), threadpool = shared_threadpool[])
12-
13-
# leakyrelu(x::AA1, a = oftype(x/1, 0.01)) =
14-
# nnp_relu_output(x, inplace ? x : similar(x), negative_slope = a, threadpool = shared_threadpool[])
15-
16-
softmax!(x::A) where A<:AbstractVecOrMat{Float64} = softmax!(Float32.(x))
9+
function softmax!(x::A) where A<:AbstractVecOrMat{Float64}
10+
x = Float32.(x)
11+
softmax!(x)
12+
end
1713

1814
softmax!(x::A) where A<:AbstractVecOrMat{Float32} =
1915
nnp_softmax_output(x, x, threadpool = shared_threadpool[])

0 commit comments

Comments
 (0)