From 4f56bd3c20b861c975342c5ee2b577d1ebe40aed Mon Sep 17 00:00:00 2001 From: Zach Koopmans Date: Tue, 10 Jun 2025 13:38:14 -0700 Subject: [PATCH] Add initial VM qualification suite Add Makefile target with the initial subset of GPU tests that qualify new VMs. The other tests in gpu-all-tests don't work on newer models and may or may not be fixed. This give a good idea of coverage of the existing tests so that more can be developed, or we can fix the gaps in existing tests. PiperOrigin-RevId: 769777107 --- Makefile | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/Makefile b/Makefile index 6aa479d4ec..3100e42d16 100644 --- a/Makefile +++ b/Makefile @@ -325,6 +325,17 @@ gpu-all-tests: gpu-images gpu-smoke-tests $(RUNTIME_BIN) @$(call sudo,test/gpu:sniffer_test,--runtime=$(RUNTIME) -test.v $(ARGS)) .PHONY: gpu-all-tests +gpu-qual-images: gpu-smoke-images load-gpu_ollama load-gpu_ollama_client load-basic_busybox load-basic_alpine load-basic_python load-gpu_nccl-tests load-gpu_cuda-tests-12-8 +.PHONY: gpu-images + +gpu-qual-tests: gpu-images gpu-smoke-tests $(RUNTIME_BIN) + @$(call install_runtime,$(RUNTIME),--nvproxy=true --nvproxy-docker=true --nvproxy-allowed-driver-capabilities=all) + @$(call sudo,test/gpu:textgen_test,--runtime=$(RUNTIME) -test.v $(ARGS)) + @$(call sudo,test/gpu:nccl_test,--runtime=$(RUNTIME) -test.v $(ARGS)) + @$(call sudo,test/gpu:sniffer_test,--runtime=$(RUNTIME) -test.v $(ARGS)) + @$(call sudo,test/gpu:cuda_12_8_test,--runtime=$(RUNTIME) -test.v $(ARGS)) +.PHONY: gpu-qual-tests + cos-gpu-all-tests: gpu-images cos-gpu-smoke-tests $(RUNTIME_BIN) @$(call install_runtime,$(RUNTIME),--nvproxy=true) @$(call sudo,test/gpu:pytorch_test,--runtime=$(RUNTIME) -test.v --cos-gpu $(ARGS))