Skip to content

Commit 2c6447b

Browse files
author
Himani Anil Deshpande
committed
[B200] Not installing NVLSM in Isolated regions
1 parent 8788fff commit 2c6447b

File tree

2 files changed

+41
-0
lines changed

2 files changed

+41
-0
lines changed

cookbooks/aws-parallelcluster-platform/resources/nvidia_nvlsm/partial/_nvidia_nvlsm_common.rb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
action :install do
1919
return unless nvlsm_installation_enabled?
20+
return if aws_region.start_with?("us-iso")
2021

2122
action_install_nvlsm_dependencies
2223
action_install_nvlsm

cookbooks/aws-parallelcluster-platform/spec/unit/resources/nvidia_nvlsm_spec.rb

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,46 @@ def self.install(chef_run)
101101
end
102102
end
103103

104+
context 'when region starts with us-iso' do
105+
let(:nvlsm_version) { "2025.03.9-1" }
106+
let(:region_name) { 'us-iso-region' }
107+
cached(:chef_run) do
108+
stubs_for_resource('nvidia_nvlsm') do |res|
109+
allow(res).to receive(:nvlsm_installation_enabled?).and_return(true)
110+
allow(res).to receive(:nvidia_enabled?).and_return(true)
111+
end
112+
runner = runner(platform: platform, version: version, step_into: ['nvidia_nvlsm']) do |node|
113+
node.override['cluster']['region'] = region_name
114+
node.override['cluster']['artifacts_s3_url'] = cluster_artifacts_s3_url
115+
node.override['cluster']['sources_dir'] = source_dir
116+
node.automatic['kernel']['machine'] = 'x86_64'
117+
end
118+
ConvergeNvidiaNvlsm.install(runner)
119+
end
120+
cached(:node) { chef_run.node }
121+
122+
before do
123+
allow_any_instance_of(Chef::RunContext).to receive(:aws_region).and_return(region_name)
124+
end
125+
126+
it 'does not install nvlsm' do
127+
is_expected.not_to run_bash("Install nvlsm")
128+
end
129+
130+
it 'does not install nvlsm dependencies' do
131+
is_expected.not_to run_bash("Install nvlsm dependencies")
132+
end
133+
134+
it 'does not download nvlsm package' do
135+
package_name = if %(redhat rocky amazon).include?(platform)
136+
"nvlsm-#{nvlsm_version}.#{arch_suffix_rhel['x86_64']}.rpm"
137+
else
138+
"nvlsm_#{nvlsm_version}_#{arch_suffix_debian['x86_64']}.deb"
139+
end
140+
is_expected.not_to create_if_missing_remote_file("#{source_dir}/#{package_name}")
141+
end
142+
end
143+
104144
%w(x86_64 aarch64).each do |arch|
105145
context "when nvlsm installation is enabled on #{arch}" do
106146
cached(:chef_run) do

0 commit comments

Comments
 (0)