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

Working torch.jit.script() and torch.compile() support #44

Merged
merged 115 commits into from
Jan 6, 2025
Merged

Working torch.jit.script() and torch.compile() support #44

merged 115 commits into from
Jan 6, 2025

Conversation

borisfom
Copy link
Collaborator

Also, restored utils.py import

mariogeiger and others added 30 commits November 20, 2024 07:10
@mariogeiger
test and quick fix for zero batch
ba9580a
@mariogeiger
trigger uniform 1d in test
0bfada9
@mariogeiger
satisfy linter
fd097c6 
Signed-off-by: Mario Geiger <[email protected]>
@mariogeiger
from typing import
251fc4d
@mariogeiger
determine math_dtype earlier
3498a32
@mariogeiger
warning with pip commands
7f3cf05
@mariogeiger
remove unused argument
2624335
@mariogeiger
changelog
91f7fce
@mariogeiger
list of inputs
4401048
@mariogeiger
add Fixed subtite
ad2db8d
@mariogeiger
Merge branch 'zero-batch' into list-inputs
dca96a8
@mariogeiger
changelog
889051a
@mariogeiger
Merge branch 'main' into list-inputs
c23816a
@mariogeiger
Merge branch 'main' into list-inputs
0487d77
@mariogeiger
add test for torch.jit.script
bc6b405
@mariogeiger
fix
c8de185
@mariogeiger
Merge branch 'list-inputs' into jit-script
5e00b37
@mariogeiger
remove keyword-only and import in the forward
16e4450
@mariogeiger
Merge branch 'main' into jit-script
e979b0f
@mariogeiger
low lvl script tests
b2c4fbb
@borisfom
TensorProduct working with script()
4669a86 
Signed-off-by: Boris Fomitchev <[email protected]>
@mariogeiger
add 4 operands tests
dc9d5b0
@borisfom
Unit tests run
334b460 
Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Restoring debug logging
79e7c5f 
Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Merge branch 'jit-script' into jit-script
46a0478
@borisfom
Merge branch 'jit-script' of github.com:NVIDIA/cuEquivariance into ji…
401fd53 
…t-script

Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Merge remote-tracking branch 'b/jit-script' into jit-script
8fce54b
@borisfom
Parameterized script test
6c5cdb0 
Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Fixed transpose for script(), script_test successful
e21c45f 
Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Fixed input mutation
779dd9c 
Signed-off-by: Boris Fomitchev <[email protected]>
borisfom and others added 18 commits December 17, 2024 14:03
@borisfom
Merge remote-tracking branch 'b/torch_trt_war' into torch_trt_war
3a7f051 
Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Merge remote-tracking branch 'b/torch_trt_war' into torch_trt_war
ef5cd76
@mariogeiger
fix strange bug
df17042
@borisfom
Merge remote-tracking branch 'b/torch_trt_war' into torch_trt_war
73647fe
@borisfom
Script fixes for uniform
a8516aa 
Signed-off-by: Boris Fomitchev <[email protected]>
@mariogeiger
add test_script_tensor_product
9a256c9
@borisfom
Moving all export tests, disabling torch_trt for now
0b37932 
Signed-off-by: Boris Fomitchev <[email protected]>
@mariogeiger
more strict input shapes
a5262bd
@mariogeiger
add back @pytest.mark.parametrize("mode", export_modes)
59dd354
@mariogeiger
fix
8a0f109
@borisfom
Fixing noconv bug
e01a7a9 
Signed-off-by: Boris Fomitchev <[email protected]>
@borisfom
Really fixing noconv
3fbadb5 
Signed-off-by: Boris Fomitchev <[email protected]>
@mariogeiger
fix linear
686782c
@mariogeiger
fix rotations
d8c1dee
@mariogeiger
fix tpfc
c837298
@mariogeiger
fix tpcw
ac2ed25
@mariogeiger
less test
64c0121
@mariogeiger
Merge branch 'main' into torch_trt_war
4e5ddc0
@mariogeiger mariogeiger self-requested a review December 19, 2024 16:31
@borisfom borisfom changed the title Torch-trt workarounds Working torch.jit.script() and torch.compile() support Dec 21, 2024
@mariogeiger mariogeiger merged commit dc2b096 into NVIDIA:main Jan 6, 2025
10 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@mariogeiger mariogeiger Awaiting requested review from mariogeiger

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@borisfom @mariogeiger