-
Notifications
You must be signed in to change notification settings - Fork 6
/
.gitlab-ci.yml
40 lines (34 loc) · 999 Bytes
/
.gitlab-ci.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
image: registry.ritc.jp/ricos/machine_learning/feml_heat:0.2.0
# variables:
# GIT_SUBMODULE_STRATEGY: recursive
default:
before_script:
- poetry config virtualenvs.create false
stages:
- test
scalar2grad:
stage: test
script:
- make differential_data PYTHON=python3.7 N_GRID_TRAIN=10 N_GRID_VALIDATION=3 N_GRID_TEST=3
- make scalar2grad PYTHON=python3.7 N_EPOCH=3
- make scalar2grad_baseline PYTHON=python3.7 N_EPOCH=3
- make scalar2grad_baseline PYTHON=python3.7 N_EPOCH=3 BASELINE_NAME=gcnii ADJ=5 INPUT=_w_node
tags:
- avx
- gpu
scalar2hessian:
stage: test
script:
- make differential_data PYTHON=python3.7 N_GRID_TRAIN=10 N_GRID_VALIDATION=3 N_GRID_TEST=3
- make scalar2hessian PYTHON=python3.7 N_EPOCH=3
- make scalar2hessian_baseline PYTHON=python3.7 N_EPOCH=3
tags:
- avx
- gpu
small_heat_nl_tensor_pipeline:
stage: test
script:
- make small_heat_nl_tensor_pipeline PYTHON=python3.7
tags:
- avx
- gpu