Skip to content

GitLab

Commit 5909accb authored by rusty1s's avatar rusty1s
Browse files

LAST TEST

parent 8dc17e4d
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
cpu/scatter.cpp
View file @ 5909accb
...@@ -4,8 +4,9 @@ ...@@ -4,8 +4,9 @@
void scatter_mul(at::Tensor src, at::Tensor index, at::Tensor out, void scatter_mul(at::Tensor src, at::Tensor index, at::Tensor out,
int64_t dim) { int64_t dim) {
int64_t elems_per_row = index.size(dim), i, idx; printf("HUHUHUHU");
printf("elems_per_row: %lli\n", elems_per_row); // int64_t elems_per_row = index.size(dim), i, idx;
// printf("elems_per_row: %lli\n", elems_per_row);
// AT_DISPATCH_ALL_TYPES(src.type(), "scatter_mul", [&] { // AT_DISPATCH_ALL_TYPES(src.type(), "scatter_mul", [&] {
// DIM_APPLY3(scalar_t, src, int64_t, index, scalar_t, out, dim, { // DIM_APPLY3(scalar_t, src, int64_t, index, scalar_t, out, dim, {
// for (i = 0; i < elems_per_row; i++) { // for (i = 0; i < elems_per_row; i++) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment