akoflow-examples/machine-learning/infrastructure-selection/workflow.yaml at main · UFFeScience/akoflow-examples · GitHub

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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
name: showcase-infrastructure-selection
spec:
  activities:
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - profile-workload
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      memoryLimit: 256Mi
      name: profile-workload
      runtime: showcase-infrastructure-selection-local
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - local-gpu
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      dependsOn:
        - profile-workload
      memoryLimit: 256Mi
      name: local-gpu
      runtime: showcase-infrastructure-selection-local
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - hpc
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      dependsOn:
        - profile-workload
      memoryLimit: 256Mi
      name: hpc
      runtime: showcase-infrastructure-selection-local
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - cloud
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      dependsOn:
        - profile-workload
      memoryLimit: 256Mi
      name: cloud
      runtime: showcase-infrastructure-selection-local
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - compare-plans
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      dependsOn:
        - local-gpu
        - hpc
        - cloud
      memoryLimit: 256Mi
      name: compare-plans
      runtime: showcase-infrastructure-selection-local
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - select-scope
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      dependsOn:
        - compare-plans
      memoryLimit: 256Mi
      name: select-scope
      runtime: showcase-infrastructure-selection-local
    - command:
        arguments:
          - /app/run_showcase.py
          - --stage
          - train
          - --output
          - .
        entrypoint: python
      cpuLimit: "1"
      dependsOn:
        - select-scope
      memoryLimit: 256Mi
      name: train
      runtime: showcase-infrastructure-selection-local
  image: ghcr.io/uffescience/akoflow-examples/showcase-infrastructure-selection:v1
  namespace: showcase