Open punithsekar opened 1 week ago
fyi @saichandax
Planned to preprocess this method(pos2posemb3d), So, Currently, this feature is not required now. Will reopen if needed. Thanks
Reopening this as similar issue in SinePositionalEncoding3D class of petr model
Describe the bug Need broadcast support for ttnn.div.
To Reproduce Steps to reproduce the behavior:
@pytest.mark.parametrize("device_params", [{"l1_small_size": 24576}], indirect=True) def test_div(device): a=torch.randn(3,1) a_ttnn=ttnn.from_torch(a,layout=ttnn.TILE_LAYOUT,device=device) b=torch.randn(1,2) torch_output=a/b # output_shape = [3,2] b_ttnn=ttnn.from_torch(b,layout=ttnn.TILE_LAYOUT,device=device) ttnn_output=ttnn.div(a_ttnn,b_ttnn)
def call(self, *function_args, **function_kwargs):
Please complete the following environment information:
Additional context Used in PETR model, in pos2posemb3d method.