···

       
1
       
1
       
 
       
{

     

       
2
       
2
       
+
       
  config,

     

       
2
       
3
       
 
       
  cmake,

     

       
3
       
4
       
 
       
  cudaPackages,

     

       
4
       
5
       
 
       
  fetchFromGitHub,

     
···

       
176
       
177
       
 
       
    license = licenses.bsd3;

     

       
177
       
178
       
 
       
    maintainers = with maintainers; [ connorbaker ];

     

       
178
       
179
       
 
       
    platforms = platforms.linux;

     

       
179
       
179
       
-
       
    # g++: error: unrecognized command-line option '-mf16c'

     

       
180
       
180
       
-
       
    broken = stdenv.hostPlatform.isAarch64;

     

       
180
       
180
       
+
       
    badPlatforms = [

     

       
181
       
181
       
+
       
      # g++: error: unrecognized command-line option '-mf16c'

     

       
182
       
182
       
+
       
      lib.systems.inspect.patterns.isAarch64

     

       
183
       
183
       
+
       
    ];

     

       
184
       
184
       
+
       
    # Requires torch.cuda._is_compiled() == True to build

     

       
185
       
185
       
+
       
    broken = !config.cudaSupport;

     

       
181
       
186
       
 
       
  };

     

       
182
       
187
       
 
       
})