blob: 9a5547a731cf21bd081e5d9f623fe2306548312f (
plain)
| 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 | require 'requirement'
# There are multiple implementations of MPI-2 available.
# http://www.mpi-forum.org/
# This requirement is used to find an appropriate one.
class MPIDependency < Requirement
  attr_reader :lang_list
  fatal true
  default_formula 'open-mpi'
  env :userpaths
  # This method must accept varargs rather than an array for
  # backwards compatibility with formulae that call it directly.
  def initialize(*tags)
    @non_functional = []
    @unknown_langs = []
    @lang_list = [:cc, :cxx, :f77, :f90] & tags
    tags -= @lang_list
    super(tags)
  end
  def mpi_wrapper_works? compiler
    compiler = which compiler
    return false if compiler.nil? or not compiler.executable?
    # Some wrappers are non-functional and will return a non-zero exit code
    # when invoked for version info.
    #
    # NOTE: A better test may be to do a small test compilation a la autotools.
    quiet_system compiler, '--version'
  end
  def inspect
    "#<#{self.class.name}: #{name.inspect} #{tags.inspect} lang_list=#{@lang_list.inspect}>"
  end
  satisfy do
    @lang_list.each do |lang|
      case lang
      when :cc, :cxx, :f90, :f77
        compiler = 'mpi' + lang.to_s
        @non_functional << compiler unless mpi_wrapper_works? compiler
      else
        @unknown_langs << lang.to_s
      end
    end
    @unknown_langs.empty? and @non_functional.empty?
  end
  env do
    # Set environment variables to help configure scripts find MPI compilers.
    # Variable names taken from:
    # https://www.gnu.org/software/autoconf-archive/ax_mpi.html
    @lang_list.each do |lang|
      compiler = 'mpi' + lang.to_s
      mpi_path = which compiler
      # Fortran 90 environment var has a different name
      compiler = 'MPIFC' if lang == :f90
      ENV[compiler.upcase] = mpi_path
    end
  end
end
 |