Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Lightning-AI/pytorch-lightning
/ endpoints
Endpoints
362 in github.com/Lightning-AI/pytorch-lightning
⨍
Functions
7,826
◇
Types & classes
1,239
↳
Endpoints
362
Route
_consolidate
click.option( "--output_file", type=click.Path(exists=True), default=None, hel
src/lightning/fabric/cli.py:None
Route
_run
click.option( "--strategy", type=click.Choice(_get_supported_strategies()), default=No
src/lightning/fabric/cli.py:None
Route
_update
torch.inference_mode() # in case `length_fn` or `batch_size_fn` computes grads
src/lightning/pytorch/callbacks/throughput_monitor.py:None
Route
example_input_array
example_input_array.setter
src/lightning/pytorch/core/module.py:None
Route
ping
app.get("/ping")
src/lightning/pytorch/serve/servable_module_validator.py:None
Route
serve
app.post("/serve")
src/lightning/pytorch/serve/servable_module_validator.py:None
Route
test_accelerator_invalid_type_devices
mock.patch("lightning.fabric.accelerators.cuda.num_cuda_devices", return_value=2)
tests/tests_fabric/test_connector.py:None
Route
test_accelerator_set_when_using_tpu
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_pytorch/models/test_tpu.py:None
Route
test_amp_ddp
mock.patch.dict( os.environ, { "CUDA_VISIBLE_DEVICES": "0,1", "SLURM_NTASKS": "2",
tests/tests_pytorch/plugins/test_amp_plugins.py:None
Route
test_amp_gpu_ddp_slurm_managed
mock.patch.dict( os.environ, { "SLURM_NTASKS": "1", "SLURM_NTASKS_PER_NODE": "1",
tests/tests_pytorch/models/test_amp.py:None
Route
test_append_metrics_file
mock.patch( # Mock the existence check, so we can simulate appending to the metrics file "lightning.fa
tests/tests_fabric/loggers/test_csv.py:None
Route
test_append_metrics_file
mock.patch( # Mock the existence check, so we can simulate appending to the metrics file "lightning.fa
tests/tests_pytorch/loggers/test_csv.py:None
Route
test_apply_batch_transfer_handler
mock.patch( "lightning.pytorch.strategies.Strategy.lightning_module", new_callable=PropertyMock, )
tests/tests_pytorch/models/test_hooks.py:None
Route
test_attach_model_callbacks
patch("lightning.pytorch.trainer.connectors.callback_connector._RICH_AVAILABLE", False)
tests/tests_pytorch/trainer/connectors/test_callback_connector.py:None
Route
test_attributes_from_environment_variables
mock.patch.dict( os.environ, { "MASTER_ADDR": "1.2.3.4", "MASTER_PORT": "500",
tests/tests_fabric/plugins/environments/test_torchelastic.py:None
Route
test_attributes_from_environment_variables
mock.patch.dict( os.environ, { "SLURM_NODELIST": "1.1.1.1, 1.1.1.2", "SLURM_JOB_ID": "
tests/tests_fabric/plugins/environments/test_slurm.py:None
Route
test_attributes_from_environment_variables
mock.patch.dict(os.environ, {"MASTER_ADDR": "1.2.3.4", "MASTER_PORT": "500", "LOCAL_RANK": "2", "NODE_RANK": "
tests/tests_fabric/plugins/environments/test_lightning.py:None
Route
test_attributes_from_environment_variables
mock.patch.dict( os.environ, { "KUBERNETES_PORT": "tcp://127.0.0.1:443", "MASTER_ADDR"
tests/tests_fabric/plugins/environments/test_kubeflow.py:None
Route
test_attributes_from_environment_variables
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_fabric/plugins/environments/test_xla.py:None
Route
test_auto_device_count
mock.patch("lightning.fabric.accelerators.cuda.num_cuda_devices", return_value=2)
tests/tests_fabric/accelerators/test_cuda.py:None
Route
test_auto_parameters_tying_tpus
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_pytorch/accelerators/test_xla.py:None
Route
test_auto_parameters_tying_tpus_nested_module
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_pytorch/accelerators/test_xla.py:None
Route
test_auto_requeue_array_job
mock.patch("lightning.pytorch.trainer.connectors.signal_connector.call")
tests/tests_pytorch/trainer/connectors/test_signal_connector.py:None
Route
test_auto_requeue_invalid_job_id
mock.patch("lightning.pytorch.trainer.connectors.signal_connector.call")
tests/tests_pytorch/trainer/connectors/test_signal_connector.py:None
Route
test_auto_requeue_job
mock.patch("lightning.pytorch.trainer.connectors.signal_connector.call")
tests/tests_pytorch/trainer/connectors/test_signal_connector.py:None
Route
test_backward_count_simple
patch("torch.Tensor.backward")
tests/tests_pytorch/trainer/optimization/test_backward_calls.py:None
Route
test_backward_count_with_closure
patch("torch.Tensor.backward")
tests/tests_pytorch/trainer/optimization/test_backward_calls.py:None
Route
test_backward_count_with_grad_accumulation
patch("torch.Tensor.backward")
tests/tests_pytorch/trainer/optimization/test_backward_calls.py:None
Route
test_backward_required
mock.patch("lightning.fabric.accelerators.mps.MPSAccelerator.is_available", return_value=False)
tests/tests_fabric/test_fabric.py:None
Route
test_broadcast_on_tpu
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_fabric/strategies/test_xla.py:None
Route
test_can_prepare_data
mock.patch("lightning.pytorch.trainer.trainer.Trainer.node_rank", new_callable=PropertyMock)
tests/tests_pytorch/core/test_datamodules.py:None
Route
test_check_for_bad_cuda_fork
mock.patch("torch.cuda.is_initialized", return_value=True)
tests/tests_fabric/strategies/launchers/test_multiprocessing.py:None
Route
test_check_for_bad_cuda_fork_with_is_in_bad_fork
mock.patch("torch.cuda._is_in_bad_fork", return_value=True)
tests/tests_fabric/strategies/launchers/test_multiprocessing.py:None
Route
test_checkpoint_callbacks_are_last
patch("lightning.pytorch.trainer.connectors.callback_connector._RICH_AVAILABLE", False)
tests/tests_pytorch/trainer/connectors/test_callback_connector.py:None
Route
test_child_process_observer
mock.patch("lightning.fabric.strategies.launchers.subprocess_script.os.kill")
tests/tests_fabric/strategies/launchers/test_subprocess_script.py:None
Route
test_collect_rng_states_if_cuda_init_fails
mock.patch("lightning.fabric.utilities.seed.torch.cuda.is_available", Mock(return_value=False))
tests/tests_fabric/utilities/test_seed.py:None
Route
test_collective_calls_with_created_group
pytest.mark.parametrize( ("fn_name", "kwargs", "return_key"), [ ("send", {"tensor": PASSED_TEN
tests/tests_fabric/plugins/collectives/test_torch_collective.py:None
Route
test_collective_manages_default_group
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/plugins/collectives/test_torch_collective.py:None
Route
test_comet_epoch_logging
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_experiment_is_still_alive_after_training_complete
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_log_graph
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_log_hyperparams
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_logger_experiment_name
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_logger_online
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_metrics_safe
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_comet_version
mock.patch.dict(os.environ, {})
tests/tests_pytorch/loggers/test_comet.py:None
Route
test_consolidate
mock.patch("lightning.fabric.cli._process_cli_args")
tests/tests_fabric/test_cli.py:None
Route
test_correct_seed_with_environment_variable
mock.patch.dict(os.environ, {"PL_GLOBAL_SEED": "2020"}, clear=True)
tests/tests_fabric/utilities/test_seed.py:None
Route
test_cpu_slurm_save_load
mock.patch("lightning.fabric.plugins.environments.slurm.SLURMEnvironment.detect", return_value=True)
tests/tests_pytorch/models/test_cpu.py:None
Route
test_cuda_accelerator_can_not_run_on_system
mock.patch("lightning.fabric.accelerators.cuda.num_cuda_devices", return_value=0)
tests/tests_fabric/test_connector.py:None
Route
test_custom_accelerator
mock.patch.dict( os.environ, { "SLURM_NTASKS": "2", "SLURM_NTASKS_PER_NODE": "1",
tests/tests_fabric/test_connector.py:None
Route
test_custom_accelerator
mock.patch.dict( os.environ, { "SLURM_NTASKS": "2", "SLURM_NTASKS_PER_NODE": "1",
tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py:None
Route
test_custom_cluster_environment_in_slurm_environment
mock.patch.dict( os.environ, { "SLURM_NTASKS": "2", "SLURM_JOB_NAME": "SOME_NAME",
tests/tests_fabric/test_connector.py:None
Route
test_custom_cluster_environment_in_slurm_environment
mock.patch.dict( os.environ, { "SLURM_NTASKS": "2", "SLURM_NTASKS_PER_NODE": "1",
tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py:None
Route
test_data_parallel_setup_module
mock.patch("lightning.fabric.strategies.dp.DataParallel")
tests/tests_fabric/strategies/test_dp.py:None
Route
test_dataloader_batch_size_updated_on_failure
patch("lightning.pytorch.tuner.batch_size_scaling.is_oom_error", return_value=True)
tests/tests_pytorch/tuner/test_scale_batch_size.py:None
Route
test_dataloaders_passed_to_fit
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_pytorch/models/test_tpu.py:None
Route
test_datamodule_parameter
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_pytorch/tuner/test_lr_finder.py:None
Route
test_ddp_barrier_non_consecutive_device_ids
mock.patch("torch.distributed.barrier")
tests/tests_pytorch/strategies/test_ddp_integration.py:None
Route
test_ddp_extra_kwargs
mock.patch("lightning.fabric.strategies.ddp.DistributedDataParallel")
tests/tests_fabric/strategies/test_ddp.py:None
Route
test_ddp_fork_on_unsupported_platform
mock.patch( "lightning.fabric.connector.torch.multiprocessing.get_all_start_methods", return_value=[],
tests/tests_fabric/test_connector.py:None
Route
test_ddp_fork_on_unsupported_platform
mock.patch( "lightning.pytorch.trainer.connectors.accelerator_connector.torch.multiprocessing.get_all_star
tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py:None
Route
test_ddp_torch_dist_is_available_in_setup
mock.patch("torch.cuda.set_device")
tests/tests_pytorch/strategies/test_ddp_integration.py:None
Route
test_deepspeed_auto_batch_size_config_select
mock.patch("deepspeed.init_distributed", autospec=True)
tests/tests_pytorch/strategies/test_deepspeed.py:None
Route
test_deepspeed_engine_is_steppable
mock.patch("deepspeed.DeepSpeedEngine", autospec=True)
tests/tests_fabric/plugins/precision/test_deepspeed.py:None
Route
test_deepspeed_env_variables_on_platforms
mock.patch("deepspeed.init_distributed", autospec=True)
tests/tests_fabric/strategies/test_deepspeed_integration.py:None
Route
test_deepspeed_load_checkpoint_one_deepspeed_engine_required
mock.patch("lightning.fabric.strategies.deepspeed._is_deepspeed_checkpoint", return_value=True)
tests/tests_fabric/strategies/test_deepspeed.py:None
Route
test_deepspeed_load_checkpoint_optimzer_state_requested
mock.patch("lightning.fabric.strategies.deepspeed._is_deepspeed_checkpoint", return_value=True)
tests/tests_fabric/strategies/test_deepspeed.py:None
Route
test_deepspeed_load_checkpoint_state_updated_with_client_state
mock.patch("lightning.fabric.strategies.deepspeed._is_deepspeed_checkpoint", return_value=True)
tests/tests_fabric/strategies/test_deepspeed.py:None
Route
test_deepspeed_precision_choice
mock.patch("lightning.fabric.accelerators.mps.MPSAccelerator.is_available", return_value=False)
tests/tests_fabric/plugins/precision/test_deepspeed_integration.py:None
Route
test_deepspeed_setup_module
mock.patch("deepspeed.initialize")
tests/tests_fabric/strategies/test_deepspeed.py:None
Route
test_deepspeed_strategy_env_variables
mock.patch("deepspeed.init_distributed", autospec=True)
tests/tests_pytorch/strategies/test_deepspeed.py:None
Route
test_deepspeed_summary_with_rich_model_summary
mock.patch("rich.table.Table.add_row", autospec=True)
tests/tests_pytorch/utilities/test_deepspeed_model_summary.py:None
Route
test_default_attributes
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/plugins/environments/test_torchelastic.py:None
Route
test_default_attributes
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/plugins/environments/test_slurm.py:None
Route
test_default_attributes
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/plugins/environments/test_lightning.py:None
Route
test_default_attributes
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/plugins/environments/test_mpi.py:None
Route
test_default_attributes
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/plugins/environments/test_kubeflow.py:None
Route
test_default_attributes
mock.patch.dict(os.environ, os.environ.copy(), clear=True)
tests/tests_fabric/plugins/environments/test_xla.py:None
Route
test_default_checkpoint_freq
mock.patch("torch.save")
tests/tests_pytorch/checkpointing/test_checkpoint_callback_frequency.py:None
Route
test_default_seed
mock.patch.dict(os.environ, clear=True)
tests/tests_fabric/utilities/test_seed.py:None
Route
test_deterministic_init
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py:None
Route
test_device_id_passed_for_cuda_devices
mock.patch("torch.distributed.init_process_group")
tests/tests_fabric/strategies/test_ddp.py:None
Route
test_device_id_passed_for_cuda_devices_pytorch
mock.patch("torch.distributed.init_process_group")
tests/tests_pytorch/strategies/test_ddp.py:None
Route
test_device_stats_cpu
mock.patch("lightning.pytorch.accelerators.cpu.get_cpu_stats", side_effect=get_cpu_stats)
tests/tests_pytorch/callbacks/test_device_stats_monitor.py:None
Route
test_device_stats_monitor_tpu
pytest.mark.xfail(raises=RuntimeError, reason="`xm.get_memory_info` is not implemented with PJRT")
tests/tests_pytorch/callbacks/test_device_stats_monitor.py:None
Route
test_devices_from_environment
mock.patch("lightning.fabric.accelerators.cuda.num_cuda_devices", return_value=8)
tests/tests_fabric/test_connector.py:None
Route
test_disable_possible_user_warnings_from_environment
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/utilities/test_warnings.py:None
Route
test_disable_possible_user_warnings_from_environment
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_pytorch/utilities/test_warnings.py:None
Route
test_dtensor_model_summary
mock.patch("lightning.pytorch.utilities.model_summary.model_summary._is_dtensor", return_value=True)
tests/tests_pytorch/utilities/test_model_summary.py:None
Route
test_empty_lsb_djob_rankfile
mock.patch.dict(os.environ, {"LSB_DJOB_RANKFILE": "", "LSB_JOBID": "1234"})
tests/tests_fabric/plugins/environments/test_lsf.py:None
Route
test_eval_step_logging
mock.patch("lightning.pytorch.loggers.TensorBoardLogger.log_metrics")
tests/tests_pytorch/trainer/logging_/test_eval_loop_logging.py:None
Route
test_example_input_array_types
pytest.mark.parametrize( ("example_input", "expected_size"), [ ([], UNKNOWN_SIZE), ((1
tests/tests_pytorch/utilities/test_model_summary.py:None
Route
test_exception_when_no_tpu_found
mock.patch("lightning.fabric.accelerators.xla._using_pjrt", return_value=True)
tests/tests_pytorch/models/test_tpu.py:None
Route
test_fabric_module_forward_conversion
pytest.mark.parametrize( ("precision", "input_type", "expected_type", "accelerator", "device_str"), [
tests/tests_fabric/test_wrappers.py:None
Route
test_fallback_from_ddp_spawn_to_ddp_on_cluster
mock.patch("lightning.fabric.plugins.environments.lsf.LSFEnvironment._read_hosts", return_value=["node0", "nod
tests/tests_fabric/test_connector.py:None
Route
test_force_nvml_based_cuda_check
mock.patch.dict(os.environ, {}, clear=True)
tests/tests_fabric/accelerators/test_cuda.py:None
Route
test_forking_on_unsupported_platform
mock.patch("lightning.fabric.strategies.launchers.multiprocessing.mp.get_all_start_methods", return_value=[])
tests/tests_fabric/strategies/launchers/test_multiprocessing.py:None
Route
test_fsdp_unsupported_on_cpu
mock.patch("lightning.fabric.accelerators.mps.MPSAccelerator.is_available", return_value=False)
tests/tests_fabric/test_connector.py:None
Route
test_get_available_flops_cuda_mapping_exists
pytest.mark.parametrize( "device_name", [ # Hopper "NVIDIA H200 SXM1", "NVIDIA
tests/tests_fabric/utilities/test_throughput.py:None
next →
1–100 of 362, ranked by callers