Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Enable torch.compile for loss_fn #597

Merged
merged 16 commits into from
Dec 8, 2023
Merged

Enable torch.compile for loss_fn #597

merged 16 commits into from
Dec 8, 2023

Commits on Dec 5, 2023

  1. compile loss fn for all workloads

    @BoyuanFeng
    BoyuanFeng committed Dec 5, 2023
    Configuration menu
    Copy the full SHA
    dc7e966 View commit details
    Browse the repository at this point in the history

Commits on Dec 6, 2023

  1. reduce eval_batch_size by half for criteo1tb_pytorch

    @BoyuanFeng
    BoyuanFeng committed Dec 6, 2023
    Configuration menu
    Copy the full SHA
    fa411fd View commit details
    Browse the repository at this point in the history

  2. further reduce eval_batch_size by half for criteo1tb_pytorch

    @BoyuanFeng
    BoyuanFeng committed Dec 6, 2023
    Configuration menu
    Copy the full SHA
    dfb8ceb View commit details
    Browse the repository at this point in the history

  3. reduce to 1/4

    @BoyuanFeng
    BoyuanFeng committed Dec 6, 2023
    Configuration menu
    Copy the full SHA
    30062f8 View commit details
    Browse the repository at this point in the history

  4. reduce eval_batch_size for criteo1tb

    @BoyuanFeng
    BoyuanFeng committed Dec 6, 2023
    Configuration menu
    Copy the full SHA
    fa3dddd View commit details
    Browse the repository at this point in the history

  5. recover eval_batch_size for criteo1tb

    @BoyuanFeng
    BoyuanFeng committed Dec 6, 2023
    Configuration menu
    Copy the full SHA
    1ddfa09 View commit details
    Browse the repository at this point in the history

  6. temporary commenting out workload.eval_model for criteo1tb

    @BoyuanFeng
    BoyuanFeng committed Dec 6, 2023
    Configuration menu
    Copy the full SHA
    bfe74ce View commit details
    Browse the repository at this point in the history

Commits on Dec 7, 2023

  1. disable torch.compile for criteo1tb

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    7a112f5 View commit details
    Browse the repository at this point in the history

  2. recover evaluation

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    4b53433 View commit details
    Browse the repository at this point in the history

  3. comment out eval to avoid tokenizer for librispeech

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    c4029ac View commit details
    Browse the repository at this point in the history

  4. recover eval_model

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    ea72827 View commit details
    Browse the repository at this point in the history

  5. comment out eval

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    a57759b View commit details
    Browse the repository at this point in the history

  6. specify workloads that benefits from compiling loss_fn

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    7132021 View commit details
    Browse the repository at this point in the history

  7. enable evaluation

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    bcf1ca0 View commit details
    Browse the repository at this point in the history

  8. nit

    @BoyuanFeng
    BoyuanFeng committed Dec 7, 2023
    Configuration menu
    Copy the full SHA
    33ecf8e View commit details
    Browse the repository at this point in the history

Commits on Dec 8, 2023

  1. Merge branch 'dev' into feat/pt2-compile-loss-fn

    @BoyuanFeng
    BoyuanFeng authored Dec 8, 2023
    Configuration menu
    Copy the full SHA
    a134d08 View commit details
    Browse the repository at this point in the history