Skip to content

[Relax][PyTorch] Add PReLU Op Support for Exported Program and FX graph #17816

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
Merged
merged 11 commits into from
Apr 18, 2025
Merged

[Relax][PyTorch] Add PReLU Op Support for Exported Program and FX graph #17816

merged 11 commits into from
Apr 18, 2025

Conversation

Deivanayaki-S
Copy link
Contributor

This PR adds support for the PReLU operation in both the Exported Program and FX graph Relax frontend pipeline.

@Deivanayaki-S Deivanayaki-S marked this pull request as ready for review April 9, 2025 08:21
Hzfengsy
Hzfengsy approved these changes Apr 9, 2025
View reviewed changes
Copy link
Member

@Hzfengsy Hzfengsy left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Hzfengsy
Copy link
Member

Hzfengsy commented Apr 9, 2025

Please address the conflict as the previous PR just has been merged :)

@Deivanayaki-S
Copy link
Contributor Author

@tvm-bot rerun

2 similar comments
@Deivanayaki-S
Copy link
Contributor Author

@tvm-bot rerun

@Deivanayaki-S
Copy link
Contributor Author

@tvm-bot rerun

@github-actions GitHub Actions
Copy link
Contributor

Failed to re-run CI in https://github.com/apache/tvm/actions/runs/14373842802

Traceback (most recent call last):
  File "ci/scripts/github/github_tvmbot.py", line 591, in comment_failure
    raise item
  File "ci/scripts/github/github_tvmbot.py", line 697, in run
    pr.rerun_jenkins_ci()
  File "ci/scripts/github/github_tvmbot.py", line 550, in rerun_jenkins_ci
    post(url, auth=("tvm-bot", TVM_BOT_JENKINS_TOKEN))
  File "/home/runner/work/tvm/tvm/ci/scripts/jenkins/git_utils.py", line 53, in post
    with request.urlopen(req, data) as response:
  File "/usr/lib/python3.8/urllib/request.py", line 222, in urlopen
    return opener.open(url, data, timeout)
  File "/usr/lib/python3.8/urllib/request.py", line 531, in open
    response = meth(req, response)
  File "/usr/lib/python3.8/urllib/request.py", line 640, in http_response
    response = self.parent.error(
  File "/usr/lib/python3.8/urllib/request.py", line 569, in error
    return self._call_chain(*args)
  File "/usr/lib/python3.8/urllib/request.py", line 502, in _call_chain
    result = func(*args)
  File "/usr/lib/python3.8/urllib/request.py", line 649, in http_error_default
    raise HTTPError(req.full_url, code, msg, hdrs, fp)
urllib.error.HTTPError: HTTP Error 502: Bad Gateway

with response

<html>
<head><title>502 Bad Gateway</title></head>
<body>
<center><h1>502 Bad Gateway</h1></center>
</body>
</html>

@yongwww
Copy link
Member

yongwww commented Apr 10, 2025

@Deivanayaki-S Please resolve the conflict so we can prioritize merging it. Seems the recent merged pr introduced the conflict

@Deivanayaki-S
Copy link
Contributor Author

@yongwww thanks for pointing it out, the conflict has been resolved.

@yongwww yongwww merged commit 103e54b into apache:main Apr 18, 2025
13 of 15 checks passed
@ysh329 ysh329 mentioned this pull request Jul 16, 2025
Closed
ShiboXing pushed a commit to ShiboXing/tvm that referenced this pull request Aug 10, 2025
@Deivanayaki-S @ShiboXing
[Relax][PyTorch] Add PReLU Op Support for Exported Program and FX gra…
47e13fc 
…ph (apache#17816)

* prelu op support and test script added

* end-of-file issue fixed

* trailing whitespace issue fixed

* fixing lint issues

* fix assertion error in test_op_nn.py file

* add test script in test_frontend_nn_op.py

* include wrapper function for prelu in op.py

* fixing unity check issue by modifying test func

* conflicts resolved

* add doc for prelu op axis arg

* fixed failing checks issue

---------

Co-authored-by: deivanayakisankaralingam <deiva@Deivanayaki>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@yongwww yongwww yongwww approved these changes

@Hzfengsy Hzfengsy Hzfengsy approved these changes

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

Successfully merging this pull request may close these issues.
3 participants
@Deivanayaki-S @Hzfengsy @yongwww