Synthetic Task Graphs for Parla Testing and Benchmarks
This is a small directory for setting up toy synthetic graphs for the Parla Python tasking system.
Task graphs execute with optional data movement and burn time with user specified busy waits (on CPU or GPU)
- scripts to generate test graphs (see
graphs/) - execute test graphs with Parla
- Enable with no data movement, lazy movement, or eager movement
- visualize test graphs dependencies (with or without data movement tasks)
- analyze test graphs (average width and critical path length)
- verify parla tasks and data movement executed in a valid ordering
- Analyze Parla output and policy
- Visualize execution order and assigned mapping
- Compute critical path time (using observed task times and bandwidth)
- Compute total data movement cost
- IN PROGRESS: Add support for slicing
You must compile the Cython sleep module which requires Cython & CUDA and must be set to an architecture newer than Pascal.
#While in this directory:
#Put nvcc_wrapper on the PATH
export PATH=$(pwd):$PATH
#Build
python setup.py install <the usual options> #(or just pip install .)
The architecture is currently hardcoded in setup.py for Frontera's RTX nodes (sm_70), but this can be changed on L24.
After updating architeture the module can be compiled simply with make or
python setup.py install <the usual options>
For simplicity we use the Kokkos nvcc wrapper to avoid having to precompile the CUDA into a shared object file.
Depending on your enviornment you may need to change the path in nvcc_wrapper to your g++ executable.
Keep in mind that your CUDA env has a maximum supported version of GCC.
The synthetic directory is currently maintained to be used as a module in-place.
It can likely be pip installed, but its not recommended.
NOTE: GPU BUSY SLEEP requires knowing the steady-state GPU clock frequency. This module is configured for Frontera. If running on a different system this frequency can be estimated with estimate_frequency(). And example of this is commented out in run.py.
Graphs are specified with "*.gph" files with the following semantics:
The first line of the file specifies an initial data partitioning. This is a comma separated list of the number of elements in each partition.
Each subsequent line represents a task.
<task_id> | <task time (μs)>, <1/vcus>, <location>, <# GIL access>, <GIL time (μs)> | <task dependencies> | <in_data> : <out_data> : <inout_data>
<task_id> can be an arbitrarily long list of comma seperated integers.
<task_dependencies> can be an arbitrarily long list of <task_ids> separated by colons ':'.
<location> can be 0=cpu, 1=gpu, 2=either.
Tasks are launched in file-order from top to bottom. Note this must be a valid launch order for Parla, otherwise the dependencies won't resolve.
As an example the following specifies a serial chain graph such that each task depends (redundently) on its two immediate predecessors.
- Each task is launched on the GPU and consumes a whole device (
vcus=1). - Each task busy waits for 50000 microseconds without the GIL and 200 microsecoonds with the GIL.
- Each task reads data_block[1] of size 100 and read/writes to data_block[0] of size 40.
40, 100
0 | 50000, 1, 1, 1, 200 | | 1 : : 0
1 | 50000, 1, 1, 1, 200 | 0 | 1 : : 0
2 | 50000, 1, 1, 1, 200 | 1 : 0 | 1 : : 0
3 | 50000, 1, 1, 1, 200 | 2 : 1 | 1 : : 0
4 | 50000, 1, 1, 1, 200 | 3 : 2 | 1 : : 0
5 | 50000, 1, 1, 1, 200 | 4 : 3 | 1 : : 0
6 | 50000, 1, 1, 1, 200 | 5 : 4 | 1 : : 0
7 | 50000, 1, 1, 1, 200 | 6 : 5 | 1 : : 0
8 | 50000, 1, 1, 1, 200 | 7 : 6 | 1 : : 0
9 | 50000, 1, 1, 1, 200 | 8 : 7 | 1 : : 0
Scripts to generate graph files are in the graphs/ directory.
viz.py is used to draw graphs to the screen (uses and REQUIRES networkx [https://networkx.org/] )
NOTE: viz.py also computes the average width and critical path length. (which can be used to bound runtime)
run.py is used to launch the tasks
'verify.py' can take the verbose standard output of 'run.py' and verify that the observed ordering is correct
Note: graph analysis should only be interpreted for runtime estimates when run without data movement edges.
The command line options for all scripts can be inspected by using '-h'. Theres a lot more than have been listed here and its worth taking a look. Important ones are: '-graph' to read a graph file, and '--data_move' to disable movement entirely, or enable lazy/eager copies.
Example usage is:
//Generate the graph
python graph/generate_serial_graph.py
//Run the graph with Parla
python run.py -graph graphs/serial.gph -data_move 0 --verbose > output.txt
//Verify that output ran in a valid ordering
python verify.py -graph graphs/serial.gph -input output.txt
//Vizualize and analyze the graph with data movement edges
python viz.py -graph graphs/serial.gph -data 1
This would generate a serial graph (with default configuration) and runs it without any data movement.
Example output on serial graph with Lazy data movement (Task size: 0.0502, Data Size: (N=2^23, d=3))
move=(1)
dim=(2)
=Task (0,) moved Data[0] from Device[-1]. Block=[1.0] | Value=[0.0], <1.0>
+Task (0,) running on Device[0] GPU for 95991043 total cycles
-Task (0,) elapsed: [0.0883781109996562] seconds
=Task (1,) moved Data[1] from Device[-1]. Block=[2.0] | Value=[-1.0], <2.0>
+Task (1,) running on Device[0] GPU for 95991043 total cycles
-Task (1,) elapsed: [0.0703934120001577] seconds
=Task (2,) moved Data[2] from Device[-1]. Block=[3.0] | Value=[-2.0], <3.0>
+Task (2,) running on Device[0] GPU for 95991043 total cycles
-Task (2,) elapsed: [0.0686743469996145] seconds
=Task (3,) moved Data[3] from Device[-1]. Block=[4.0] | Value=[-3.0], <4.0>
+Task (3,) running on Device[0] GPU for 95991043 total cycles
-Task (3,) elapsed: [0.06101883400060615] seconds
=Task (4,) moved Data[4] from Device[-1]. Block=[5.0] | Value=[-4.0], <5.0>
+Task (4,) running on Device[0] GPU for 95991043 total cycles
-Task (4,) elapsed: [0.060688632000164944] seconds
=Task (5,) moved Data[5] from Device[-1]. Block=[6.0] | Value=[-5.0], <6.0>
+Task (5,) running on Device[0] GPU for 95991043 total cycles
-Task (5,) elapsed: [0.06036870899970381] seconds
=Task (6,) moved Data[6] from Device[-1]. Block=[7.0] | Value=[-6.0], <7.0>
+Task (6,) running on Device[0] GPU for 95991043 total cycles
-Task (6,) elapsed: [0.06025352800043038] seconds
=Task (7,) moved Data[7] from Device[-1]. Block=[8.0] | Value=[-7.0], <8.0>
+Task (7,) running on Device[0] GPU for 95991043 total cycles
-Task (7,) elapsed: [0.060560630000509263] seconds
=Task (8,) moved Data[8] from Device[-1]. Block=[9.0] | Value=[-8.0], <9.0>
+Task (8,) running on Device[0] GPU for 95991043 total cycles
-Task (8,) elapsed: [0.06069639100041968] seconds
=Task (9,) moved Data[9] from Device[-1]. Block=[10.0] | Value=[-9.0], <10.0>
+Task (9,) running on Device[0] GPU for 95991043 total cycles
-Task (9,) elapsed: [0.060309198000140896] seconds
Elapsed Internal Main Task: 0.6859247080001296 seconds
Total Elapsed: 1.7873143400001936 seconds
Time to Spawn Main Task: 0.00023081200015440118 seconds
Task Ordering: VALID
Data Blocks: VALID
Data Movement: VALID
Graph analysis:
--------------
The longest path in the DAG is: 10
Generation Sizes. Min: 1, Mean: 1.0, Max: 1
Assuming equal sized tasks and no data movement:
Degree of Parallelism in Generational Ordering = 1
Average Task Size: 0.0502 seconds
Lower bound estimate: 0.502 seconds
Expected Serial Time: 0.502 seconds
