Commit1875dcc

authored and

committed

Removed arbritrary ROCM ifdef that governs threads.

Signed-off-by: Bryce Ferenczi <frenzi@hotmail.com.au>

1 parent14f0cd7 commit1875dccCopy full SHA for 1875dcc

File tree

-4

lines changed

-4

lines changed

Lines changed: 0 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -146,11 +146,7 @@ Tensor& binary_cross_entropy_backward_out_cuda(const Tensor& grad, const Tensor&`
`146`	`146`	`namespace {`
`147`	`147`
`148`	`148`	`intnll_loss_threads(int64_t nframe){`
`149`		`-#if defined(USE_ROCM)`
`150`	`149`	`returnstd::clamp(1 <<static_cast<int64_t>(std::round(std::log2(nframe/16))),32,1024);`
`151`		`-#else`
`152`		`-return32;`
`153`		`-#endif`
`154`	`150`	`}`
`155`	`151`
`156`	`152`	// NOTE(crcrpar): `Byte` support was added for https://github.com/pytorch/pytorch/issues/59765.

Comments

(0)