Skip to content

Commit

Permalink
remove comment
Browse files Browse the repository at this point in the history
  • Loading branch information
zyy-martin committed Dec 18, 2024
1 parent 1dde877 commit 7290336
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions test/cpp/test_aten_xla_tensor_2.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2120,19 +2120,12 @@ TEST_F(AtenXlaTensorTest, TestCumProdCastLong) {
TEST_F(AtenXlaTensorTest, TestCumMax) {
torch::Tensor input = torch::rand({4, 3, 4});
int rank = input.dim();
LOG(INFO) << "input: " << input;
for (int dim = -rank; dim < rank; ++dim) {
std::tuple<torch::Tensor, torch::Tensor> result = torch::cummax(input, dim);
LOG(INFO) << "torch::cummax: [values]: " << std::get<0>(result)
<< " [indices]: " << std::get<1>(result);
ForEachDevice([&](const torch::Device& device) {
LOG(INFO) << "device: " << device;
torch::Tensor xla_input = CopyToDevice(input, device);
std::tuple<torch::Tensor, torch::Tensor> xla_result =
torch::cummax(xla_input, dim);
LOG(INFO) << "xla_input: " << xla_input;
LOG(INFO) << "xla_result: [values]: " << std::get<0>(xla_result)
<< " [indices]: " << std::get<1>(xla_result);
AllClose(std::get<0>(result), std::get<0>(xla_result));
AllClose(std::get<1>(result), std::get<1>(xla_result));
});
Expand Down

0 comments on commit 7290336

Please sign in to comment.