Commit 1c2c154

committed

Merge remote-tracking branch 'origin/main' into wengshiy/embeddingbag_krnl

2 parents 623285e + df7bf37 commit 1c2c154Copy full SHA for 1c2c154

284 files changed

+16254

-4845

lines changed

.github
- pytorch-probot.yml
- scripts/torchao_model_releases
  - quantize_and_upload.py
  - release.sh
- workflows
README.md
benchmarks
- _models
  - eval_hf_models.py
- benchmark_aq.py
- dashboard
  - ci_microbenchmark_runner.py
  - microbenchmark_quantization_config.yml
- float8
- inference
  - bench_float8_inference.py
- microbenchmarks
- mx_formats
  - cast_bench.py
- prototype
  - blockwise_fp8_training
    - bench_1x128_128x128_gemms.py
    - bench_1x128_128x1_gemms.py
  - moe_training
docs/source
examples
- sam2_amg_server
  - compile_export_utils.py
- sam2_vos_example
  - compile_export_utils.py
scripts
- quick_start.py
setup.py
test
torchao
- _executorch_ops.py
- _models
  - llama
    - eval.py
    - generate.py
  - sam
    - eval_combo.py
- core
  - config.py
- csrc
  - cuda/mx_kernels
    - mxfp8_quantize.cuh
  - rocm/swizzle
    - swizzle.cpp
- dtypes
- experimental
  - CMakeLists.txt
  - benchmark_infra/ios
    - output_redirect.mm
  - kernels/cpu
    - aarch64
      - CMakeLists.txt
      - bitpacking
        bitpack.h
      - embedding
        embedding_lut.h
      - tests
        CMakeLists.txt
        build_and_run_tests.sh
        test_bitpack_fallback_compatibility.cpp
        test_bitpacking.cpp
        test_embedding_lut.cpp
        test_utils.h
    - fallback
      - CMakeLists.txt
      - bitpacking
        bitpack.h
        uint1.h
        uint2.h
        uint3.h
        uint4.h
        uint5.h
        uint6.h
        uint7.h
      - tests
        CMakeLists.txt
        build_and_run_tests.sh
        test_bitpacking.cpp
  - op_lib.py
  - ops
    - embedding_lut
    - groupwise_lowbit_weight_lut
      - op_groupwise_lowbit_weight_lut-impl.h
      - op_groupwise_lowbit_weight_lut_aten.cpp
    - packed_weights_header.h
    - tests
      - CMakeLists.txt
      - build_and_run_tests.sh
  - tests
- float8
- kernel
- ops.py
- optim
- prototype
  - autoround
    - eval_autoround.py
  - blockwise_fp8_training
    - kernels.py
  - float8nocompile
    - examples
      - example.py
    - test
      - fsdp_test.py
      - train_test.py
  - hqq
    - hqq_tinygemm_linear.py
  - moe_training
  - mx_formats
  - parq
    - optim
      - quantopt.py
    - quant
      - quant_api.py
      - uniform_torchao.py
    - utils.py
  - quantization
    - autoquant_v2.py
    - codebook_coreml
    - codebook_groupwise
    - codebook_utils
      - __init__.py
      - codebook_utils.py
    - dynamic_activation_lut
      - int8_dynamic_activation_lut_tensor.py
    - gguf
      - gguf_quantized_tensor.py
  - spinquant
    - hadamard_utils.py
- quantization
- sparsity
  - sparse_api.py
  - training
    - __init__.py
    - autograd.py
- testing
  - model_architectures.py
  - pt2e
    - utils.py
  - training
    - roofline_utils.py
  - utils.py
- utils.py
tutorials/quantize_vit
- run_vit_b_quant.py

284 files changed

+16254

-4845

lines changed

`‎.github/pytorch-probot.yml‎`

-Original file line number
+Diff line change
 - ciflow/benchmark
 - ciflow/tutorials
 - ciflow/rocm
 +- ciflow/4xh100

Comments

(0)