12630: first try at adding GPU support, no tests yet

[arvados.git] / lib / crunchrun / executor.go

diff --git a/lib/crunchrun/executor.go b/lib/crunchrun/executor.go
index 65bf7427b9601c465fb21d811c5cb79d2d41a0f8..bffd701bcd4d0246841a2dfbb4e2d9cde5b9c2ab 100644 (file)
--- a/lib/crunchrun/executor.go
+++ b/lib/crunchrun/executor.go
@@ -24,6 +24,7 @@ type containerSpec struct {
        BindMounts    map[string]bindmount
        Command       []string
        EnableNetwork bool
+       EnableCUDA    bool
        NetworkMode   string // docker network mode, normally "default"
        CgroupParent  string
        Stdin         io.Reader
@@ -58,4 +59,7 @@ type containerExecutor interface {
 
        // Release resources (temp dirs, stopped containers)
        Close()
+
+       // Name of runtime engine ("docker", "singularity")
+       Runtime() string
 }