diff options
Diffstat (limited to 'infra/bots/gen_tasks.go')
-rw-r--r-- | infra/bots/gen_tasks.go | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/infra/bots/gen_tasks.go b/infra/bots/gen_tasks.go index dafd403720..53e68db2bd 100644 --- a/infra/bots/gen_tasks.go +++ b/infra/bots/gen_tasks.go @@ -991,6 +991,10 @@ func test(b *specs.TasksCfgBuilder, name string, parts map[string]string, compil recipe := "test" if strings.Contains(name, "SKQP") { recipe = "skqp_test" + } else if strings.Contains(name, "OpenCL") { + // TODO(dogben): Longer term we may not want this to be called a "Test" task, but until we start + // running hs_bench or kx, it will be easier to fit into the current job name schema. + recipe = "compute_test" } extraProps := map[string]string{} iid := internalHardwareLabel(parts) @@ -1291,6 +1295,12 @@ func process(b *specs.TasksCfgBuilder, name string) { pkgs = append(pkgs, b.MustGetCipdPackageFromAsset("linux_vulkan_intel_driver_debug")) } } + if strings.Contains(name, "OpenCL") { + pkgs = append(pkgs, + b.MustGetCipdPackageFromAsset("opencl_ocl_icd_linux"), + b.MustGetCipdPackageFromAsset("opencl_intel_neo_linux"), + ) + } } if strings.Contains(name, "ProcDump") { pkgs = append(pkgs, b.MustGetCipdPackageFromAsset("procdump_win")) |