Updates floor division to use floor division operator

Mike Ruberry · facebook-github-bot · commit fad3cf076984 · 2020-06-03T09:46:42.000-07:00
Summary:
Performing floor division with torch.div is deprecated and will soon throw a runtime error. This diff updates the floor division to use the floor division operator.

Created from Diffusion's 'Open in Editor' feature.

Reviewed By: myleott

Differential Revision: D21848480

fbshipit-source-id: c9374e9406f4ba388f30315294eee7a2a4fcfecc
diff --git a/fairseq/search.py b/fairseq/search.py
@@ -335,7 +335,7 @@ def step(self, step: int, lprobs, scores):
             k,
         )
 
-        final_beams = torch.div(final_indices, k)
+        final_beams = final_indices // k
 
         for i in range(bsz):
             final_indices[i] = indices[i][final_indices[i]]

Original file line number	Diff line number	Diff line change
`@@ -335,7 +335,7 @@ def step(self, step: int, lprobs, scores):`
`335`	`335`	`k,`
`336`	`336`	`)`
`337`	`337`
`338`		`- final_beams = torch.div(final_indices, k)`
	`338`	`+ final_beams = final_indices // k`
`339`	`339`
`340`	`340`	`for i in range(bsz):`
`341`	`341`	`final_indices[i] = indices[i][final_indices[i]]`