Làm thế nào để có được số lõi trên một máy với OCaml?

Question

Tôi đang song song một số công việc trong chương trình OCaml của mình (với parmap) nhưng tôi không muốn mã hóa cứng số lượng lõi vào ứng dụng của mình. Có cách nào để lấy số lõi trong thời gian chạy không? Tôi không muốn thêm bất kỳ phụ thuộc nào nữa (không có gì ngoài parmap hoặc JS của core). Tôi có cảm giác tôi đang tìm kiếm một số cuộc gọi đơn giản trong stdlib ...

EDIT: không cần phải di chuyển. Làm việc trên Linux là đủ tốt.

Answer 1

I once có cùng một câu hỏi. Đó là những gì tôi cuối cùng đi kèm với (tôi không muốn ràng buộc C):

let cpu_count() = 
    try match Sys.os_type with 
    | "Win32" -> int_of_string (Sys.getenv "NUMBER_OF_PROCESSORS") 
    | _ -> 
     let i = Unix.open_process_in "getconf _NPROCESSORS_ONLN" in 
     let close() = ignore (Unix.close_process_in i) in 
     try Scanf.fscanf i "%d" (fun n -> close(); n) with e -> close(); raise e 
    with 
    | Not_found | Sys_error _ | Failure _ | Scanf.Scan_failure _ 
    | End_of_file | Unix.Unix_error (_, _, _) -> 1 

Nếu bạn không muốn Unix bạn có thể thay thế open_process_in bởi một văn bản Sys.command vào một tập tin tạm thời. Thử nghiệm trên linux và osx, báo cáo để làm việc trên mingw nhưng không phải trên Cygwin vào thời điểm đó.

Cập nhật. Lưu ý rằng điều này không hoạt động trên freebsd, như đã đề cập here bạn cần sử dụng sysctl -n hw.ncpu. Tuy nhiên, kể từ khi Sys.os_type không có mức chi tiết phù hợp, bạn cần phải điều kiện hóa kết quả của uname -s bất cứ khi nào Sys.os_type khác với Win32.