Merge pull request #374 from jgehrcke/jp/do-not-lock-prepare-unprepare

Configure DRA lib to send (un)prepare calls concurrently

Author: jgehrcke

cmd/compute-domain-kubelet-plugin/driver.go

@@ -70,6 +70,13 @@ func NewDriver(ctx context.Context, config *Config) (*driver, error) {
 		kubeletplugin.KubeClient(driver.client),
 		kubeletplugin.NodeName(config.flags.nodeName),
 		kubeletplugin.DriverName(DriverName),
+		// By default, the DRA library serializes (un)prepare calls. That is, at
+		// most one such call is exposed to the driver at any given time.
+		// Disable that behavior: this driver has codependent prepare() actions
+		// (where for the first prepare() to eventually complete, a second
+		// prepare() must be incoming). Concurrency management for incoming
+		// requests is done with this driver's work queue abstraction.
+		kubeletplugin.Serialize(false),
 	)
 	if err != nil {
 		return nil, err

cmd/gpu-kubelet-plugin/driver.go

@@ -52,6 +52,7 @@ func NewDriver(ctx context.Context, config *Config) (*driver, error) {
 		kubeletplugin.KubeClient(driver.client),
 		kubeletplugin.NodeName(config.flags.nodeName),
 		kubeletplugin.DriverName(DriverName),
+		kubeletplugin.Serialize(false),
 	)
 	if err != nil {
 		return nil, err