Skip to content

Commit b5241a5

Browse files
cocoa-xucocoa-xu
committed
mix format
1 parent d5379c5 commit b5241a5

File tree

2 files changed

+153
-50
lines changed

2 files changed

+153
-50
lines changed

‎mix.exs

+70-7
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ defmodule Mix.Tasks.Compile.EvisionPrecompiled do
8383
if enable_cuda do
8484
{defalt_cuda, default_cudnn} = Metadata.default_cuda_version()
8585
System.put_env("EVISION_ENABLE_CUDA", "true")
86+
8687
{
8788
System.get_env("EVISION_CUDA_VERSION", defalt_cuda),
8889
System.get_env("EVISION_CUDNN_VERSION", default_cudnn)
@@ -92,7 +93,15 @@ defmodule Mix.Tasks.Compile.EvisionPrecompiled do
9293
{"", ""}
9394
end
9495

95-
get_download_url(target, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version)
96+
get_download_url(
97+
target,
98+
version,
99+
nif_version,
100+
enable_contrib,
101+
enable_cuda,
102+
cuda_version,
103+
cudnn_version
104+
)
96105
end
97106

98107
def checksum_file(app \\ Mix.Project.config()[:app]) when is_atom(app) do
@@ -532,9 +541,26 @@ defmodule Mix.Tasks.Compile.EvisionPrecompiled do
532541
"evision-nif_#{nif_version}-#{target}-contrib-cuda#{cuda_version}-cudnn#{cudnn_version}-#{version}#{with_ext}"
533542
end
534543

535-
def get_download_url(target, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version) do
544+
def get_download_url(
545+
target,
546+
version,
547+
nif_version,
548+
enable_contrib,
549+
enable_cuda,
550+
cuda_version,
551+
cudnn_version
552+
) do
536553
tar_file =
537-
filename(target, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version, ".tar.gz")
554+
filename(
555+
target,
556+
version,
557+
nif_version,
558+
enable_contrib,
559+
enable_cuda,
560+
cuda_version,
561+
cudnn_version,
562+
".tar.gz"
563+
)
538564

539565
"#{Metadata.github_url()}/releases/download/v#{version}/#{tar_file}"
540566
end
@@ -574,11 +600,38 @@ defmodule Mix.Tasks.Compile.EvisionPrecompiled do
574600
Path.join([build_path, "lib", "#{app}", "priv"])
575601
end
576602

577-
def prepare(target, os, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version) do
578-
name = filename(target, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version)
603+
def prepare(
604+
target,
605+
os,
606+
version,
607+
nif_version,
608+
enable_contrib,
609+
enable_cuda,
610+
cuda_version,
611+
cudnn_version
612+
) do
613+
name =
614+
filename(
615+
target,
616+
version,
617+
nif_version,
618+
enable_contrib,
619+
enable_cuda,
620+
cuda_version,
621+
cudnn_version
622+
)
579623

580624
filename =
581-
filename(target, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version, ".tar.gz")
625+
filename(
626+
target,
627+
version,
628+
nif_version,
629+
enable_contrib,
630+
enable_cuda,
631+
cuda_version,
632+
cudnn_version,
633+
".tar.gz"
634+
)
582635

583636
cache_dir = cache_dir()
584637
cache_file = Path.join([cache_dir, filename])
@@ -817,7 +870,17 @@ defmodule Mix.Tasks.Compile.EvisionPrecompiled do
817870
{default_cuda_version, default_cudnn_version} = Metadata.default_cuda_version()
818871
cuda_version = System.get_env("EVISION_CUDA_VERSION", default_cuda_version)
819872
cudnn_version = System.get_env("EVISION_CUDNN_VERSION", default_cudnn_version)
820-
prepare(target, os, version, nif_version, enable_contrib, enable_cuda, cuda_version, cudnn_version)
873+
874+
prepare(
875+
target,
876+
os,
877+
version,
878+
nif_version,
879+
enable_contrib,
880+
enable_cuda,
881+
cuda_version,
882+
cudnn_version
883+
)
821884
else
822885
_ ->
823886
raise RuntimeError, "Cannot use precompiled binaries."

‎test/evision_mat_test.exs

+83-43
Original file line numberDiff line numberDiff line change
@@ -9,29 +9,45 @@ defmodule Evision.Mat.Test do
99
shape = {3, 4, 3}
1010
mat = Evision.Mat.zeros(shape, :u8)
1111
zeros = Nx.tensor(0, type: :u8) |> Nx.broadcast(shape)
12-
assert 1 == Nx.to_number(Nx.all_close(
13-
zeros,
14-
Evision.Mat.to_nx(mat, {Nx.BinaryBackend, []})
15-
))
12+
13+
assert 1 ==
14+
Nx.to_number(
15+
Nx.all_close(
16+
zeros,
17+
Evision.Mat.to_nx(mat, {Nx.BinaryBackend, []})
18+
)
19+
)
1620

1721
expected_results = Nx.tensor(1, type: :u8) |> Nx.broadcast(shape)
1822
result = Evision.add(mat, Nx.tensor([1], type: :u8))
19-
assert 1 == Nx.to_number(Nx.all_close(
20-
expected_results,
21-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
22-
))
23+
24+
assert 1 ==
25+
Nx.to_number(
26+
Nx.all_close(
27+
expected_results,
28+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
29+
)
30+
)
2331

2432
result = Evision.add(mat, Nx.tensor(1, type: :u8))
25-
assert 1 == Nx.to_number(Nx.all_close(
26-
expected_results,
27-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
28-
))
33+
34+
assert 1 ==
35+
Nx.to_number(
36+
Nx.all_close(
37+
expected_results,
38+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
39+
)
40+
)
2941

3042
result = Evision.add(mat, 1)
31-
assert 1 == Nx.to_number(Nx.all_close(
32-
expected_results,
33-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
34-
))
43+
44+
assert 1 ==
45+
Nx.to_number(
46+
Nx.all_close(
47+
expected_results,
48+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
49+
)
50+
)
3551
end
3652

3753
@tag :nx
@@ -41,10 +57,14 @@ defmodule Evision.Mat.Test do
4157
result = Evision.add(mat_val, Nx.tensor(tensor_val, type: :f16))
4258

4359
expected_results = Nx.tensor(mat_val + tensor_val, type: :f16)
44-
assert 1 == Nx.to_number(Nx.all_close(
45-
expected_results,
46-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
47-
))
60+
61+
assert 1 ==
62+
Nx.to_number(
63+
Nx.all_close(
64+
expected_results,
65+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
66+
)
67+
)
4868
end
4969

5070
@tag :nx
@@ -58,10 +78,14 @@ defmodule Evision.Mat.Test do
5878
result = Evision.add(mat_val, mat_val)
5979

6080
expected_results = Nx.tensor([[2], [2], [2], [2]], type: :f64)
61-
assert 1 == Nx.to_number(Nx.all_close(
62-
expected_results,
63-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
64-
))
81+
82+
assert 1 ==
83+
Nx.to_number(
84+
Nx.all_close(
85+
expected_results,
86+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
87+
)
88+
)
6589
end
6690

6791
@tag :nx
@@ -75,10 +99,14 @@ defmodule Evision.Mat.Test do
7599
result = Evision.add(mat_val, {1, 2.2, 3.3})
76100

77101
expected_results = Nx.tensor([[2.0], [3.2], [4.3], [1]], type: :f64)
78-
assert 1 == Nx.to_number(Nx.all_close(
79-
expected_results,
80-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
81-
))
102+
103+
assert 1 ==
104+
Nx.to_number(
105+
Nx.all_close(
106+
expected_results,
107+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
108+
)
109+
)
82110
end
83111

84112
@tag :nx
@@ -88,13 +116,17 @@ defmodule Evision.Mat.Test do
88116
# [7.],
89117
# [9.0],
90118
# [0.]])
91-
result = Evision.add({1,2,3}, {4,5,6})
119+
result = Evision.add({1, 2, 3}, {4, 5, 6})
92120

93121
expected_results = Nx.tensor([[5], [7], [9], [0]], type: :f64)
94-
assert 1 == Nx.to_number(Nx.all_close(
95-
expected_results,
96-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
97-
))
122+
123+
assert 1 ==
124+
Nx.to_number(
125+
Nx.all_close(
126+
expected_results,
127+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
128+
)
129+
)
98130
end
99131

100132
@tag :nx
@@ -106,13 +138,17 @@ defmodule Evision.Mat.Test do
106138
# [4.],
107139
# [6.],
108140
# [4.]])
109-
result = Evision.add({1,2,3}, {1,2,3,4})
141+
result = Evision.add({1, 2, 3}, {1, 2, 3, 4})
110142

111143
expected_results = Nx.tensor([[2], [4], [6], [4]], type: :f64)
112-
assert 1 == Nx.to_number(Nx.all_close(
113-
expected_results,
114-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
115-
))
144+
145+
assert 1 ==
146+
Nx.to_number(
147+
Nx.all_close(
148+
expected_results,
149+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
150+
)
151+
)
116152
end
117153

118154
@tag :nx
@@ -125,13 +161,17 @@ defmodule Evision.Mat.Test do
125161
# [7.],
126162
# [8.]
127163
# [9.]])
128-
result = Evision.add({1,2,3}, {4,5,6,7,8})
164+
result = Evision.add({1, 2, 3}, {4, 5, 6, 7, 8})
129165

130166
expected_results = Nx.tensor([[5], [6], [7], [8], [9]], type: :f64)
131-
assert 1 == Nx.to_number(Nx.all_close(
132-
expected_results,
133-
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
134-
))
167+
168+
assert 1 ==
169+
Nx.to_number(
170+
Nx.all_close(
171+
expected_results,
172+
Evision.Mat.to_nx(result, {Nx.BinaryBackend, []})
173+
)
174+
)
135175
end
136176
end
137177

0 commit comments

Comments
 (0)