Skip to content

Enable partial support for MacOS/MPS training - #626

Open
dsocolobsky wants to merge 2 commits into
mainfrom
dy/enable-macos-training
Open

Enable partial support for MacOS/MPS training#626
dsocolobsky wants to merge 2 commits into
mainfrom
dy/enable-macos-training

Merge branch 'main' into dy/enable-macos-training

32acdd6
Select commit
Loading
Failed to load commit list.
Garnix CI / check workspace-test-ci [x86_64-linux] succeeded Mar 18, 2026 in 4m 26s

check workspace-test-ci [x86_64-linux]

check workspace-test-ci [x86_64-linux] succeeded

Details

Last 100 lines of logs:

psyche-workspace-nextest-0.2.0>         PASS [   0.708s] (136/222) psyche-modeling distro::tests::test_decompress_1d
psyche-workspace-nextest-0.2.0>         PASS [   0.728s] (137/222) psyche-data-provider::weighted test_weighted_data_provider_unequal_weights
psyche-workspace-nextest-0.2.0>         PASS [   0.732s] (138/222) psyche-modeling distro::tests::test_decode_1d
psyche-workspace-nextest-0.2.0>         PASS [   0.748s] (139/222) psyche-data-provider::weighted test_weighted_data_provider_auto_weights
psyche-workspace-nextest-0.2.0>         PASS [   0.752s] (140/222) psyche-data-provider::preprocessed loads_hermes3_subset
psyche-workspace-nextest-0.2.0>         PASS [   0.753s] (141/222) psyche-data-provider weighted::http::export_bindings_weightedhttpprovidersconfig
psyche-workspace-nextest-0.2.0>         PASS [   0.794s] (142/222) psyche-data-provider::weighted test_weighted_data_provider_consistency
psyche-workspace-nextest-0.2.0>         PASS [   0.739s] (143/222) psyche-modeling distro::tests::test_encode_2d
psyche-workspace-nextest-0.2.0>         PASS [   0.843s] (144/222) psyche-modeling distro::tests::test_dct
psyche-workspace-nextest-0.2.0>         PASS [   0.855s] (145/222) psyche-modeling distro::tests::test_decompress_2d
psyche-workspace-nextest-0.2.0>         PASS [   0.706s] (146/222) psyche-modeling distro::tests::test_get_prime_divisors
psyche-workspace-nextest-0.2.0>         PASS [   0.849s] (147/222) psyche-modeling distro::tests::test_get_divisors
psyche-workspace-nextest-0.2.0>         PASS [   0.984s] (148/222) psyche-data-provider::local loads_fineweb_subset
psyche-workspace-nextest-0.2.0>         PASS [   1.027s] (149/222) psyche-data-provider::local loads_dolma_subset
psyche-workspace-nextest-0.2.0>         PASS [   1.099s] (150/222) psyche-modeling distro::tests::test_1bit_matches_non_quant
psyche-workspace-nextest-0.2.0>         PASS [   0.955s] (151/222) psyche-modeling distro::tests::test_get_smaller_split
psyche-workspace-nextest-0.2.0>         PASS [   1.187s] (152/222) psyche-data-provider::http test_http_data_provider
psyche-workspace-nextest-0.2.0>         PASS [   1.385s] (153/222) psyche-data-provider weighted::http::tests::test_http_multi_data_provider
psyche-workspace-nextest-0.2.0>         PASS [   0.892s] (154/222) psyche-modeling distro::tests::test_idct
psyche-workspace-nextest-0.2.0>         PASS [   0.885s] (155/222) psyche-network download::scheduler::tests::test_config_retries_dont_consume_capacity
psyche-workspace-nextest-0.2.0>         PASS [   1.544s] (156/222) psyche-data-provider::http test_http_data_provider_shuffled
psyche-workspace-nextest-0.2.0>         PASS [   0.822s] (157/222) psyche-network download::scheduler::tests::test_waiters_are_served_fifo
psyche-workspace-nextest-0.2.0>         PASS [   0.012s] (158/222) psyche-solana-authorizer test_id
psyche-workspace-nextest-0.2.0>         PASS [   0.012s] (159/222) psyche-solana-distributor test_id
psyche-workspace-nextest-0.2.0>         PASS [   0.940s] (160/222) psyche-modeling distro::tests::test_signed_vals_reconstructs_original_sign
psyche-workspace-nextest-0.2.0>         PASS [   0.024s] (161/222) psyche-solana-coordinator test_id
psyche-workspace-nextest-0.2.0>         PASS [   0.830s] (162/222) psyche-network serializable_tensor::tests::test_roundtrip_bool_tensor1d
psyche-workspace-nextest-0.2.0>         PASS [   0.014s] (163/222) psyche-solana-mining-pool test_id
psyche-workspace-nextest-0.2.0>         PASS [   0.017s] (164/222) psyche-solana-tooling::mod suites::memnet_coordinator_data_layout::run
psyche-workspace-nextest-0.2.0>         PASS [   0.895s] (165/222) psyche-network download::scheduler::tests::test_wait_for_capacity_errors_on_actor_shutdown
psyche-workspace-nextest-0.2.0>         PASS [   0.970s] (166/222) psyche-network download::scheduler::tests::test_distro_retries_dont_consume_capacity
psyche-workspace-nextest-0.2.0>         PASS [   0.976s] (167/222) psyche-network download::scheduler::tests::test_parameter_retries_are_immediate
psyche-workspace-nextest-0.2.0>         PASS [   1.010s] (168/222) psyche-network download::scheduler::tests::test_distro_retry_not_immediately_due
psyche-workspace-nextest-0.2.0>         PASS [   1.058s] (169/222) psyche-network download::scheduler::tests::test_capacity_grants_up_to_max
psyche-workspace-nextest-0.2.0>         PASS [   1.064s] (170/222) psyche-modeling distro::tests::test_large_numbers
psyche-workspace-nextest-0.2.0>         PASS [   1.044s] (171/222) psyche-network download::scheduler::tests::test_distro_max_retries_exceeded
psyche-workspace-nextest-0.2.0>         PASS [   1.007s] (172/222) psyche-network download::scheduler::tests::test_release_unblocks_waiter
psyche-workspace-nextest-0.2.0>         PASS [   1.020s] (173/222) psyche-network download::scheduler::tests::test_model_sharing_retry_respects_capacity
psyche-workspace-nextest-0.2.0>         PASS [   1.013s] (174/222) psyche-network download::scheduler::tests::test_remove_retry
psyche-workspace-nextest-0.2.0>         PASS [   0.017s] (175/222) psyche-solana-treasurer test_id
psyche-workspace-nextest-0.2.0>         PASS [   1.008s] (176/222) psyche-network serializable_tensor::tests::test_roundtrip_bool_tensor2d
psyche-workspace-nextest-0.2.0>         PASS [   1.089s] (177/222) psyche-network download::scheduler::tests::test_model_sharing_never_exceeds_max_retries
psyche-workspace-nextest-0.2.0>         PASS [   0.869s] (178/222) psyche-network serializable_tensor::tests::test_roundtrip_tensor1d
psyche-workspace-nextest-0.2.0>         PASS [   1.806s] (179/222) psyche-centralized-testing::integration_tests connect_single_node
psyche-workspace-nextest-0.2.0>         PASS [   0.209s] (180/222) psyche-solana-tooling::mod suites::memnet_coordinator_init_free::run
psyche-workspace-nextest-0.2.0>         PASS [   1.055s] (181/222) psyche-network router::tests::test_shutdown
psyche-workspace-nextest-0.2.0>         PASS [   0.961s] (182/222) psyche-network serializable_tensor::tests::test_roundtrip_bool_tensor_single_element
psyche-workspace-nextest-0.2.0>         PASS [   1.148s] (183/222) psyche-network download::scheduler::tests::test_distro_retries_returned_and_removed
psyche-workspace-nextest-0.2.0>         PASS [   0.900s] (184/222) psyche-network serializable_tensor::tests::test_roundtrip_tensor2d
psyche-workspace-nextest-0.2.0>         PASS [   0.230s] (185/222) psyche-solana-tooling::mod suites::memnet_mining_pool_cap_deposits::run
psyche-workspace-nextest-0.2.0>         PASS [   0.353s] (186/222) psyche-solana-tooling::mod suites::memnet_authorizer_full_cycle::run
psyche-workspace-nextest-0.2.0>         PASS [   0.962s] (187/222) psyche-network serialized_distro::tests::test_roundtrip_distro_result_1bit
psyche-workspace-nextest-0.2.0>         PASS [   1.123s] (188/222) psyche-network serializable_tensor::tests::test_roundtrip_bool_tensor_unusual_shape
psyche-workspace-nextest-0.2.0>         PASS [   0.321s] (189/222) psyche-solana-tooling::mod suites::memnet_mining_pool_live_freeze::run
psyche-workspace-nextest-0.2.0>         PASS [   1.216s] (190/222) psyche-network serializable_tensor::tests::test_roundtrip_bool_tensor_non_divisible_by_8
psyche-workspace-nextest-0.2.0>         PASS [   0.358s] (191/222) psyche-solana-tooling::mod suites::memnet_treasurer_create_claim::run
psyche-workspace-nextest-0.2.0>         PASS [   0.371s] (192/222) psyche-solana-tooling::mod suites::memnet_mining_pool_full_cycle::run
psyche-workspace-nextest-0.2.0>         PASS [   0.385s] (193/222) psyche-solana-tooling::mod suites::memnet_treasurer_create_update::run
psyche-workspace-nextest-0.2.0>         PASS [   0.982s] (194/222) psyche-network tcp::tests::test_challenge_rejects_invalid_signature
psyche-workspace-nextest-0.2.0>         PASS [   0.922s] (195/222) psyche-network tcp::tests::test_multiple_clients
psyche-workspace-nextest-0.2.0>         PASS [   0.826s] (196/222) psyche-network tcp::tests::test_tcp_handshake_uses_only_iroh_key
psyche-workspace-nextest-0.2.0>         PASS [   1.072s] (197/222) psyche-network tcp::tests::test_message_exchange_after_iroh_handshake
psyche-workspace-nextest-0.2.0>         PASS [   0.628s] (198/222) psyche-solana-tooling::mod suites::memnet_distributor_many_claims::run
psyche-workspace-nextest-0.2.0>         PASS [   2.349s] (199/222) psyche-modeling distro::tests::test_artifical_distro_results_roundtrip
psyche-workspace-nextest-0.2.0>         PASS [   0.841s] (200/222) psyche-solana-tooling::mod suites::memnet_coordinator_full_round::run
psyche-workspace-nextest-0.2.0>         PASS [   1.490s] (201/222) psyche-network serializable_tensor::tests::test_roundtrip_tensor_manyd
psyche-workspace-nextest-0.2.0>         PASS [   1.686s] (202/222) psyche-network serializable_tensor::tests::test_roundtrip_bool_tensor_manyd
psyche-workspace-nextest-0.2.0>         PASS [   0.781s] (203/222) psyche-solana-tooling::mod suites::memnet_treasurer_full_epoch::run
psyche-workspace-nextest-0.2.0>         PASS [   0.944s] (204/222) psyche-solana-tooling::mod suites::memnet_distributor_single_claim::run
psyche-workspace-nextest-0.2.0>         PASS [   1.024s] (205/222) psyche-solana-tooling::mod suites::memnet_distributor_claim_vesting::run
psyche-workspace-nextest-0.2.0>         PASS [   2.775s] (206/222) psyche-centralized-testing::integration_tests state_change_waiting_for_members_to_warmup
psyche-workspace-nextest-0.2.0>         PASS [   2.822s] (207/222) psyche-inference vllm::tests::test_list_engines
psyche-workspace-nextest-0.2.0>         PASS [   1.446s] (208/222) psyche-solana-tooling::mod suites::memnet_coordinator_rewards::run
psyche-workspace-nextest-0.2.0>         PASS [   2.337s] (209/222) psyche-network router::tests::test_allowlist
psyche-workspace-nextest-0.2.0>         PASS [   3.253s] (210/222) psyche-centralized-testing::integration_tests state_change_shutdown_node_in_warmup
psyche-workspace-nextest-0.2.0>         PASS [   3.419s] (211/222) psyche-centralized-testing::integration_tests connect_multiple_nodes
psyche-workspace-nextest-0.2.0>         PASS [  16.763s] (212/222) psyche-centralized-testing::integration_tests state_change_waiting_for_members_to_round_witness
psyche-workspace-nextest-0.2.0>         PASS [  17.329s] (213/222) psyche-centralized-testing::integration_tests replace_node_and_complete_round
psyche-workspace-nextest-0.2.0>         PASS [  19.426s] (214/222) psyche-network test::test_retry_connection
psyche-workspace-nextest-0.2.0>         PASS [  21.695s] (215/222) psyche-centralized-testing::integration_tests state_change_waiting_for_members_to_round_train
psyche-workspace-nextest-0.2.0>         PASS [  22.133s] (216/222) psyche-network test::test_retry_connection_mid_download
psyche-workspace-nextest-0.2.0>         PASS [  24.843s] (217/222) psyche-centralized-testing::integration_tests shutdown_node_in_training_and_complete_round
psyche-workspace-nextest-0.2.0>         PASS [  31.729s] (218/222) psyche-centralized-testing::integration_tests validate_all_clients_participate_in_witness_bloom
psyche-workspace-nextest-0.2.0>         PASS [  42.753s] (219/222) psyche-centralized-testing::integration_tests client_join_in_training
psyche-workspace-nextest-0.2.0>         PASS [  45.747s] (220/222) psyche-centralized-testing::integration_tests finish_epoch
psyche-workspace-nextest-0.2.0>         SLOW [> 60.000s] (-------) psyche-centralized-testing::integration_tests client_join_in_training_and_get_model_using_p2p
psyche-workspace-nextest-0.2.0>         SLOW [> 60.000s] (-------) psyche-centralized-testing::integration_tests two_clients_join_in_training_and_get_model_using_p2p
psyche-workspace-nextest-0.2.0>         PASS [  86.256s] (221/222) psyche-centralized-testing::integration_tests two_clients_join_in_training_and_get_model_using_p2p
psyche-workspace-nextest-0.2.0>         PASS [  89.164s] (222/222) psyche-centralized-testing::integration_tests client_join_in_training_and_get_model_using_p2p
psyche-workspace-nextest-0.2.0> ------------
psyche-workspace-nextest-0.2.0>      Summary [  89.164s] 222 tests run: 222 passed (2 slow), 2 skipped
psyche-workspace-nextest-0.2.0> checkPhase completed in 3 minutes 25 seconds
psyche-workspace-nextest-0.2.0> Running phase: installPhase
psyche-workspace-nextest-0.2.0> linking previous artifacts /nix/store/38icjcwrdm30a6fq2jvsayf8s1f7jf8j-psyche-workspace-deps-0.2.0/target.tar.zst to /nix/store/wdr2rqrycywiy4z40hajzyhy2hna4qq9-psyche-workspace-nextest-0.2.0/target.tar.zst
psyche-workspace-nextest-0.2.0> /*stdin*\            : 28.63%   (   853 MiB =>    244 MiB, /nix/store/wdr2rqrycywiy4z40hajzyhy2hna4qq9-psyche-workspace-nextest-0.2.0/target.tar.zst) 
psyche-workspace-nextest-0.2.0> Running phase: fixupPhase
psyche-workspace-nextest-0.2.0> shrinking RPATHs of ELF executables and libraries in /nix/store/wdr2rqrycywiy4z40hajzyhy2hna4qq9-psyche-workspace-nextest-0.2.0
psyche-workspace-nextest-0.2.0> checking for references to /build/ in /nix/store/wdr2rqrycywiy4z40hajzyhy2hna4qq9-psyche-workspace-nextest-0.2.0...
psyche-workspace-nextest-0.2.0> patching script interpreter paths in /nix/store/wdr2rqrycywiy4z40hajzyhy2hna4qq9-psyche-workspace-nextest-0.2.0
psyche-workspace-nextest-0.2.0> shrinking RPATHs of ELF executables and libraries in /nix/store/wdr2rqrycywiy4z40hajzyhy2hna4qq9-psyche-workspace-nextest-0.2.0