mirror of
https://github.com/netbirdio/gvisor.git
synced 2026-05-22 17:12:49 -07:00
Etienne PerotandgVisor bot
3649ca9d9e
Unlike Ubuntu VMs where we use Docker's `--gpus` flag, COS VMs do not use this flag and instead mount the NVIDIA library directories automatically. However, nothing guarantees that these directories are added to the LD config. This change fixes that. It take advantage of the fact that all GPU tests have the sniffer binary as entrypoint, which slightly overloads the role of the sniffer within the GPU test infrastructure... but then again the ioctl sniffer is already deeply intertwined with ld configuration because it already overrides the `ioctl` libc function, so this doesn't seem like too big of a stretch. This change makes the ffmpeg test succeed with `runc` on COS, but they still fail with gVisor (with `CUDA_ERROR_OUT_OF_MEMORY` errors). So there must be some further gVisor-specific error. Updates #11351 Updates #11321 PiperOrigin-RevId: 715222952
39 lines
1.3 KiB
Go
39 lines
1.3 KiB
Go
// Copyright 2025 The gVisor Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
)
|
|
|
|
// addPathToLd adds the given path to the ld cache.
|
|
func addPathToLd(ctx context.Context, path string) error {
|
|
const myLdConfigPath = "/etc/ld.so.conf.d/gvisor.conf"
|
|
if err := os.WriteFile(myLdConfigPath, []byte(fmt.Sprintf("# Generated by gVisor ioctl sniffer\n%s", path)), 0644); err != nil {
|
|
return fmt.Errorf("failed to write to ld config file %q: %w", myLdConfigPath, err)
|
|
}
|
|
if err := os.Remove("/etc/ld.so.cache"); err != nil && !os.IsNotExist(err) {
|
|
return fmt.Errorf("failed to remove ld cache file: %w", err)
|
|
}
|
|
output, err := exec.CommandContext(ctx, "ldconfig").CombinedOutput()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to run ldconfig: %w; output: %s", err, string(output))
|
|
}
|
|
return nil
|
|
}
|