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

Update __truediv__ comment in python bindings. #3586

Merged
merged 4 commits into from
Dec 14, 2024
Merged

Update __truediv__ comment in python bindings. #3586

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
119c323
update __truediv__ comment
rdspring1 Dec 13, 2024
db7c8b5
update
rdspring1 Dec 13, 2024
26e41c0
remove w
rdspring1 Dec 13, 2024
e1d4939
Merge branch 'main' into truediv_type
rdspring1 Dec 13, 2024
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
13 changes: 7 additions & 6 deletions csrc/python_frontend/python_bindings.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2058,14 +2058,15 @@ void initNvFuserPythonBindings(PyObject* module) {
"__rshift__", "bitwise_right_shift", bitwise_right_shift)
// In python, __truediv__ (/) always returns a float regardless of whether
// the input arguments are float or integer. __truediv__ (/) corresponds with
// pytorch torch.true_divide(a, b). The __div__ operator is depreciated in
// pytorch torch.true_divide(a, b). The __div__ operator is deprecated in
// python 3.
//
// In nvfuser, truediv function has the same semantics as python's
// operator __truediv__ (/). The div function truncates the result instead
// of promoting it to float, so it has the same semantics as the C++'s (/)
// operator. In pytorch, torch.div(a, b, rounding_mode='trunc') corresponds
// C-style integer division.
// In nvfuser, truediv function in csrc/ops/arith.h has the same semantics as
// python's operator __truediv__ (/). The div function in csrc/ops/arith.h
// truncates the result instead of promoting it to float. It has the same
// semantics as the C++'s (/) operator. In pytorch,
// torch.div(a, b, rounding_mode='trunc') corresponds C-style integerw
Copy link
Collaborator

Choose a reason for hiding this comment

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

Suggested change
// torch.div(a, b, rounding_mode='trunc') corresponds C-style integerw
// torch.div(a, b, rounding_mode='trunc') corresponds C-style integer

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

I blame vim. 😄

// division.
//
// Hence, in the python frontend, the __truediv__ (/) python operator maps to
// trunc division.
Expand Down
Loading