Skip to content

[Relax][PyTorch] Sort.default #17852

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

Merged
MasterJH5574 merged 16 commits into from
Apr 21, 2025
Merged

[Relax][PyTorch] Sort.default #17852

Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
e13fa3d
unit test
hugolatendresse Apr 14, 2025
5b23c30
full.default
hugolatendresse Apr 14, 2025
35aee29
linting
hugolatendresse Apr 14, 2025
5c0e18b
ones ok
hugolatendresse Apr 14, 2025
e5c0258
sort default - still as TODO
hugolatendresse Apr 14, 2025
81f48b8
merge main
hugolatendresse Apr 17, 2025
d8ec9a5
sort test passes
hugolatendresse Apr 17, 2025
23ee26a
cleanup
hugolatendresse Apr 17, 2025
fe53f9c
cleanup test
hugolatendresse Apr 17, 2025
18bf952
lint
hugolatendresse Apr 17, 2025
9ed717e
remove print
hugolatendresse Apr 17, 2025
68bf177
test values and indices
hugolatendresse Apr 17, 2025
6984480
Merge remote-tracking branch 'upstream/main' into sortdefault
hugolatendresse Apr 19, 2025
775c491
frontend test
hugolatendresse Apr 19, 2025
eaa4ded
Merge remote-tracking branch 'upstream/main' into sortdefault
hugolatendresse Apr 21, 2025
633827e
merge main
hugolatendresse Apr 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
linting
  • Loading branch information
@hugolatendresse
hugolatendresse committed Apr 14, 2025
commit 35aee297ba2ca01dbdf2695267cf1869b399a95b
5 changes: 3 additions & 2 deletions tests/python/relax/test_from_exported_to_cuda.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -71,13 +71,14 @@ def __init__(self):

def forward(self, x):
return torch.full((2, 3), 3.141592)

torch_module = FullModel().eval()

raw_data = np.random.rand(3,3).astype("float32")
raw_data = np.random.rand(3, 3).astype("float32")

assert_torch_output_vs_tvm_from_exported_to_cuda(raw_data, torch_module, target, dev)


@tvm.testing.parametrize_targets("cuda")
def test_tensor_clamp(target, dev):
class ClampBothTensor(torch.nn.Module):
Expand Down