You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Crash while running this test. Substituting ttnn::reciprocal with any other unary op doesn't result in crash.
TEST_F(N300UtilsTest, TestXTensorReplicateAllReduce_96_768) {
auto* device = &ttml::autograd::ctx().get_device();
auto mesh_shape = device->shape();
std::cout << "I AM HERE IN TEST 0\n";
xt::xarray<float> xtensor = xt::random::rand({96 * 768}, -0.05, 0.05).reshape({1, 1, 96, 768});
ttml::core::XTensorToMeshVariant<float> replicate_composer = ttml::core::ReplicateXTensorToMesh<float>(mesh_shape);
auto tensor = ttml::core::from_xtensor(xtensor, device, replicate_composer);
std::cout << "I AM HERE IN TEST 1\n";
tensor = ttnn::add(tensor, 100.0F);
std::cout << "I AM HERE IN TEST 2\n";
tensor = ttnn::reciprocal(tensor);
std::cout << "I AM HERE IN TEST 3\n";
}
To Reproduce
Run specified test above
See crash and prints before last print
Expected behavior
Should work without crash.
Please complete the following environment information:
Describe the bug
Crash while running this test. Substituting
ttnn::reciprocal
with any other unary op doesn't result in crash.To Reproduce
Expected behavior
Should work without crash.
Please complete the following environment information:
Additional context
There might be a deeper problem, so we would like to understand what's going on with this
random
operation. Thanks in advance.The text was updated successfully, but these errors were encountered: