···

       
65
       
65
       
 
       
let

     

       
66
       
66
       
 
       
  setBool = v: if v then "1" else "0";

     

       
67
       
67
       
 
       


     

       
68
       
68
       
-
       
  # https://github.com/pytorch/pytorch/blob/v1.13.1/torch/utils/cpp_extension.py#L1751

     

       
68
       
68
       
+
       
  # https://github.com/pytorch/pytorch/blob/v2.0.1/torch/utils/cpp_extension.py#L1744

     

       
69
       
69
       
 
       
  supportedTorchCudaCapabilities =

     

       
70
       
70
       
 
       
    let

     

       
71
       
71
       
-
       
      real = ["3.5" "3.7" "5.0" "5.2" "5.3" "6.0" "6.1" "6.2" "7.0" "7.2" "7.5" "8.0" "8.6"];

     

       
71
       
71
       
+
       
      real = ["3.5" "3.7" "5.0" "5.2" "5.3" "6.0" "6.1" "6.2" "7.0" "7.2" "7.5" "8.0" "8.6" "8.9" "9.0"];

     

       
72
       
72
       
 
       
      ptx = lists.map (x: "${x}+PTX") real;

     

       
73
       
73
       
 
       
    in

     

       
74
       
74
       
 
       
    real ++ ptx;