mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
6de6041585
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39868 ### Summary why disable NNPACK on iOS - To stay consistency with our internal version - It's currently blocking some external users due to its lack support of x86 architecture - https://github.com/pytorch/pytorch/issues/32040 - https://discuss.pytorch.org/t/undefined-symbols-for-architecture-x86-64-for-libtorch-in-swift-unit-test/84552/6 - NNPACK uses fast convolution algorithms (FFT, winograd) to reduce the computational complexity of convolutions with large kernel size. The algorithmic speedup is limited to specific conv params which are unlikely to appear in mobile networks. - Since XNNPACK has been enabled, it performs much better than NNPACK on depthwise-separable convolutions which is the algorithm being used by most of mobile computer vision networks. ### Test Plan - CI Checks Test Plan: Imported from OSS Differential Revision: D22087365 Pulled By: xta0 fbshipit-source-id: 89a959b0736c1f8703eff10723a8fbd02357fd4a
79 lines
2.8 KiB
Ruby
79 lines
2.8 KiB
Ruby
require 'xcodeproj'
|
|
require 'fileutils'
|
|
require 'optparse'
|
|
|
|
options = {}
|
|
option_parser = OptionParser.new do |opts|
|
|
opts.banner = 'Script for setting up TestApp.xcodeproj'
|
|
opts.on('-t', '--team_id ', 'development team ID') { |value|
|
|
options[:team_id] = value
|
|
}
|
|
end.parse!
|
|
puts "Current directory: #{Dir.pwd}"
|
|
install_path = File.expand_path("../../../build_ios/install")
|
|
if not Dir.exist? (install_path)
|
|
raise "path doesn't exist:#{install_path}!"
|
|
end
|
|
xcodeproj_path = File.expand_path("../TestApp.xcodeproj")
|
|
if not File.exist? (xcodeproj_path)
|
|
raise "path doesn't exist:#{xcodeproj_path}!"
|
|
end
|
|
puts "Setting up TestApp.xcodeproj..."
|
|
project = Xcodeproj::Project.open(xcodeproj_path)
|
|
targets = project.targets
|
|
header_search_path = ['$(inherited)', "#{install_path}/include"]
|
|
libraries_search_path = ['$(inherited)', "#{install_path}/lib"]
|
|
other_linker_flags = ['$(inherited)', "-all_load"]
|
|
# TestApp and TestAppTests
|
|
targets.each do |target|
|
|
target.build_configurations.each do |config|
|
|
config.build_settings['HEADER_SEARCH_PATHS'] = header_search_path
|
|
config.build_settings['LIBRARY_SEARCH_PATHS'] = libraries_search_path
|
|
config.build_settings['OTHER_LDFLAGS'] = other_linker_flags
|
|
config.build_settings['ENABLE_BITCODE'] = 'No'
|
|
dev_team_id = options[:team_id]
|
|
if dev_team_id
|
|
config.build_settings['DEVELOPMENT_TEAM'] = dev_team_id
|
|
end
|
|
end
|
|
end
|
|
puts "Installing the testing model..."
|
|
model_path = File.expand_path("./model.pt")
|
|
if not File.exist?(model_path)
|
|
raise "model.pt can't be found!"
|
|
end
|
|
config_path = File.expand_path("./config.json")
|
|
if not File.exist?(config_path)
|
|
raise "config.json can't be found!"
|
|
end
|
|
group = project.main_group.find_subpath(File.join('TestApp'),true)
|
|
group.set_source_tree('SOURCE_ROOT')
|
|
group.files.each do |file|
|
|
if (file.name.to_s.end_with?(".pt") || file.name == "config.json")
|
|
group.remove_reference(file)
|
|
targets.each do |target|
|
|
target.resources_build_phase.remove_file_reference(file)
|
|
end
|
|
end
|
|
end
|
|
model_file_ref = group.new_reference(model_path)
|
|
config_file_ref = group.new_reference(config_path)
|
|
targets.each do |target|
|
|
target.resources_build_phase.add_file_reference(model_file_ref, true)
|
|
target.resources_build_phase.add_file_reference(config_file_ref, true)
|
|
end
|
|
puts "Linking static libraries..."
|
|
libs = ['libc10.a', 'libclog.a', 'libXNNPACK.a', 'libeigen_blas.a', 'libcpuinfo.a', 'libpytorch_qnnpack.a', 'libtorch_cpu.a', 'libtorch.a']
|
|
targets.each do |target|
|
|
target.frameworks_build_phases.clear
|
|
for lib in libs do
|
|
path = "#{install_path}/lib/#{lib}"
|
|
if File.exist?(path)
|
|
libref = project.frameworks_group.new_file(path)
|
|
target.frameworks_build_phases.add_file_reference(libref)
|
|
end
|
|
end
|
|
end
|
|
|
|
project.save
|
|
puts "Done."
|