Skip to content

Comments

[ROCm] Change ortmodule test#15884

Merged
PeixuanZuo merged 1 commit intomainfrom
peixuanzuo/fix_ortmodule_test
May 31, 2023
Merged

[ROCm] Change ortmodule test#15884
PeixuanZuo merged 1 commit intomainfrom
peixuanzuo/fix_ortmodule_test

Conversation

@PeixuanZuo
Copy link
Contributor

@PeixuanZuo PeixuanZuo commented May 10, 2023

Change ortmodule test because rocm ep behaves differently than cuda.
The warning from torch The first argument to symbolic functions is deprecated in 1.13 and will be removed in the future. Please annotate treat the first argument (g) as GraphContext and use context information from the object instead. appears twice on ROCm EP.

On ROCm EP, the log is shown as below:

The first argument to symbolic functions is deprecated in 1.13 and will be removed in the future. Please annotate treat the first argument (g) as GraphContext and use context information from the object instead.
The first argument to symbolic functions is deprecated in 1.13 and will be removed in the future. Please annotate treat the first argument (g) as GraphContext and use context information from the object instead.
User Module's attribute name _torch_module collides with ORTModule's attribute name. User Module's attribute may not be returned when trying to retrieve the attribute through ORTModule.
User Module's attribute name load_state_dict collides with ORTModule's attribute name. User Module's method may not be called upon invocation through ORTModule.

@PeixuanZuo PeixuanZuo requested review from baijumeswani and ytaous May 11, 2023 03:55
PeixuanZuo added a commit that referenced this pull request May 15, 2023
update ROCm/MIGraphX CI to ROC5.5.

TODO:
two PR to fix failure on
orttraining/orttraining/test/python/orttraining_test_ortmodule_api.py
-
test_gradient_correctness_minmax/test_gradient_correctness_argmax_unfold/test_gradient_correctness_argmax_diagonal
(#15903)
- test_ortmodule_attribute_name_collision_warning
(#15884)
@PeixuanZuo PeixuanZuo requested review from cloudhan and mindest May 16, 2023 03:52
prathikr pushed a commit that referenced this pull request May 16, 2023
update ROCm/MIGraphX CI to ROC5.5.

TODO:
two PR to fix failure on
orttraining/orttraining/test/python/orttraining_test_ortmodule_api.py
-
test_gradient_correctness_minmax/test_gradient_correctness_argmax_unfold/test_gradient_correctness_argmax_diagonal
(#15903)
- test_ortmodule_attribute_name_collision_warning
(#15884)
@PeixuanZuo PeixuanZuo force-pushed the peixuanzuo/fix_ortmodule_test branch from eb8ec02 to c325715 Compare May 24, 2023 03:44
Copy link
Contributor

@mindest mindest left a comment

Choose a reason for hiding this comment

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

LGTM, thanks.

@PeixuanZuo PeixuanZuo merged commit 3dc5179 into main May 31, 2023
@PeixuanZuo PeixuanZuo deleted the peixuanzuo/fix_ortmodule_test branch May 31, 2023 07:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants