Endpoints — github.com/bitsandbytes-foundation/bitsandbytes

hub / github.com/bitsandbytes-foundation/bitsandbytes / endpoints

Endpoints78 in github.com/bitsandbytes-foundation/bitsandbytes

⨍ Functions 720◇ Types & classes 117↳ Endpoints 78

Routelinear8bit

pytest.fixture(params=get_available_devices(no_cpu=True))

tests/test_linear8bitlt.py:None

Routetest_4bit_compressed_stats

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_4bit_embedding_warnings

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_4bit_embedding_weight_fsdp_fix

pytest.mark.parametrize("device", get_available_devices(no_cpu=True))

tests/test_modules.py:None

Routetest_4bit_linear_warnings

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_4bit_linear_weight_fsdp_fix

pytest.mark.parametrize("device", get_available_devices(no_cpu=True))

tests/test_modules.py:None

Routetest_4bit_quant

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_4bit_quant_large

pytest.mark.parametrize("device", get_available_devices(no_cpu=True))

tests/test_functional.py:None

Routetest_ademamix_state_dict_no_nan

pytest.mark.parametrize("device", get_available_devices())

tests/test_optim.py:None

Routetest_blockwise_cpu_large

pytest.mark.skipif("cpu" not in get_available_devices(), reason="CPU is required")

tests/test_functional.py:None

Routetest_coo_double_quant

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_coo_int8_vectorwise_quant

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_copy_param

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_deepcopy_param

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_dequant_mm

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_dequantize_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_dequantize_blockwise

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_dequantize_blockwise_non_contiguous

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_different_blocksizes

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_dynamic_blockwise_quantization

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_dynamic_blockwise_quantization_large

pytest.mark.parametrize("device", get_available_devices(no_cpu=True))

tests/test_functional.py:None

Routetest_embedding_error

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_embedding_lossless

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_few_bit_quant

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_fp8_quant

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_fsdp_state_dict_save_4bit

pytest.mark.skipif(not get_available_devices(no_cpu=True), reason="FSDP requires an accelerator device")

tests/test_linear4bit.py:None

Routetest_gemm_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_gemm_4bit_non_float32_offset

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_gemv_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_gemv_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_gemv_eye_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_global_config

pytest.mark.parametrize("device", get_available_devices())

tests/test_optim.py:None

Routetest_gradient_behavior

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_int8_linear_matmul

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_int8_linear_matmul

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_int8_linear_matmul_half

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_int8_linear_matmul_out

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_int8_mm_dequant

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_int8_scaled_mm

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_int8_vectorwise_quant

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_integrated_int8_linear_matmul

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_kbit_backprop

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_linear4bit_torch_compile

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_linear4bit_torch_compile_activation_checkpointing

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_linear8bitlt_device_movement

pytest.mark.parametrize("device", get_available_devices(no_cpu=True))

tests/test_linear8bitlt.py:None

Routetest_linear8bitlt_inference

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_linear8bitlt_no_fp16_weights

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_linear8bitlt_torch_compile

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear8bitlt.py:None

Routetest_linear_kbit_fp32_bias

pytest.mark.parametrize("device", get_available_devices())

tests/test_modules.py:None

Routetest_linear_no_igemmlt

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear8bitlt.py:None

Routetest_linear_serialization

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear8bitlt.py:None

Routetest_linear_serialization

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_lion32bit_weight_decay

pytest.mark.parametrize("device", get_available_devices(), ids=id_formatter("device"))

tests/test_optim.py:None

Routetest_matmul_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_autograd.py:None

Routetest_matmul_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_matmul_4bit_weight_orientation

pytest.mark.parametrize("device", get_available_devices())

tests/test_functional.py:None

Routetest_matmullt

pytest.mark.parametrize("device", get_available_devices())

tests/test_autograd.py:None

Routetest_moe_parameter_shape

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_moe_realistic_forward

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_multiple_parameters

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_optimizer32bit

pytest.mark.parametrize("device", get_available_devices(), ids=id_formatter("device"))

tests/test_optim.py:None

Routetest_optimizer8bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_optim.py:None

Routetest_override_config_after_register

pytest.mark.parametrize("device", get_available_devices())

tests/test_optim.py:None

Routetest_params4bit_quant_state_attr_access

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_params4bit_real_serialization

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_params4bit_torch_chunk_split

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_prequantized_replacement

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_quant_state_preservation

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_quant_storage_shard_roundtrip

pytest.mark.parametrize("device", get_available_devices())

tests/test_linear4bit.py:None

Routetest_quantize_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_quantize_4bit_non_contiguous

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_quantize_4bit_not_divisible_by_blocksize

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_quantize_4bit_roundtrip_non_contiguous

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_quantize_blockwise

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_quantize_blockwise_non_contiguous

pytest.mark.parametrize("device", get_available_devices())

tests/test_ops.py:None

Routetest_replace_parameter_4bit

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_state_dict_functionality

pytest.mark.parametrize("device", get_available_devices())

tests/test_parametrize.py:None

Routetest_version_selection

pytest.mark.parametrize( "spec,fake_libs,hip_version,expected_name,expect_warning", [ # exact

tests/test_cuda_setup_evaluator.py:None