sharjeelyunus/github-issues-dataset huggingface dataset:
Repository: angular
Issue: :host pseudo-element does not function with combinator selectors, such as :has(+ .my-class)
Labels: area: compiler,core: stylesheets

### Which @angular/* package(s) are the source of the bug?

Don't known / other

### Is this a regression?

No

### Description

When writing scss using the :host pseudo-element, I've found that you cannot use combinator selectors within the :has() selector. For example, the following scss does not work:

:host:has(> child-element) {
//doSomething
}

However, this would normally be valid scss for any other given element.

There is a workaround, which is as follows:

::ng-deep {
    my-host-component:has(> child-element) {
        //doSomething
    }
}

However, that requires using ng-deep, which is deprecated, and generally not a great practice.

Any other solutions would be appreciated!

### Please provide a link to a minimal reproduction of the bug

https://stackblitz.com/edit/trtkcr?file=src%2Fapp%2FmyComponent.ts

### Please provide the exception or error you saw

_No response_

### Please provide the environment you discovered this bug in (run `ng version`)

Angular CLI: 18.1.3
Node: 18.20.3
Package Manager: npm 10.2.3
OS: linux x64

Angular: 18.1.2
... animations, cdk, common, compiler, compiler-cli, core, forms
... material, platform-browser

Package                      Version
------------------------------------------------------
@angular-devkit/architect    0.1801.3
@angular-devkit/core         18.1.3
@angular-devkit/schematics   18.1.3
@angular/build               18.1.3
@angular/cli                 18.1.3
@schematics/angular          18.1.3
rxjs                         7.8.1
typescript                   5.5.4
zone.js                      0.14.8

### Anything else?

I have found this issue in any version I've tested (Angular 14 and 18), but have not tested others.
Repository: neovim
Issue: virt_lines doesn't scroll horizontally
Labels: enhancement,marks

### Problem

[lsp_lines.nvim](https://git.sr.ht/~whynothugo/lsp_lines.nvim) uses `nvim_buf_set_extmark` to create extmarks in a buffer. These extmarks use `virtual_lines` with some spaces on the left so that the properly align with the position of a diagnostic on the line above.

When I scroll to the right on buffer, the extmarks remain anchored in position relative to the _window_, not to the buffer. As a result, the extmarks are not properly positioned relative to the buffer, and the overflowing portion of the extmarks are unreadable.

extmarks and their position are declared for buffers, so their position should be relative to buffers, not windows.

Given that multiple windows can render the same buffer, I cannot take the scroll offset into account when rendering the extmarks, because one the "correct" value varies per-window.

### Steps to reproduce

Add this to your `init.lua`:

```lua
vim.keymap.set("n", "<Leader>x", function()
  local ns = vim.api.nvim_create_namespace("test")
  local line = {
    "hello, this is just some test text. It should be anchored under first line",
  }
  local virt_lines = { { line } }

  vim.api.nvim_buf_set_extmark(0, ns, 0, 0, {
    virt_lines = virt_lines,
  })
end)
```

Open some file and press `250ix`, then press <kbd>Esc</kbd> and <kbd>Leader</kbd><kbd>x</kbd>.

Scroll to the right.

The virtual text will maintain its absolute position relative to the window, not relative to the buffer.

### Expected behavior

When scrolling to the right, the virtual line should scroll too, keeping its position relative to the buffer.

### Nvim version (nvim -v)

NVIM v0.10.2

### Vim (not Nvim) behaves the same?

n/a

### Operating system/version

Alpine Linux

### Terminal name/version

foot

### $TERM environment variable

foot

### Installation

Alpine packages (edge)
Repository: neovim
Issue: Netrw does not open remote file
Labels: bug-vim,netrw

### Problem

In specific cases, Netrw plugin does not open remote file. Result depends on **file name** and **order** on the command line.

### Steps to reproduce

This **works**:
`nvim -p 'scp://root@REDACTED:22//etc/network/interfaces' 'scp://root@REDACTED:22//etc/network/interfaces.new'`

This does **not**:
`nvim -p 'scp://root@REDACTED:22//etc/network/interfaces.new' 'scp://root@REDACTED:22//etc/network/interfaces'`

The only difference is **order of files**.

Further tests:

**Works**:
`nvim -p 'scp://root@REDACTED:22//tmp/bbb' 'scp://root@REDACTED:22//tmp/bbb.old'`

**Does not work**:
`nvim -p 'scp://root@REDACTED:22//tmp/bbb.old' 'scp://root@REDACTED:22//tmp/bbb'`


### Expected behavior

Neovim always opens both files.

### Nvim version (nvim -v)

NVIM v0.10.2

### Vim (not Nvim) behaves the same?

Yes, Vi IMproved 9.1

### Operating system/version

Archlinux

### Terminal name/version

st 0.9.2

### $TERM environment variable

xterm-256color

### Installation

Standard install from repo (extra)
Repository: vscode
Issue: Explore terminal quick fixes based on selection
Labels: feature-request,terminal-quick-fix,terminal-chat

Flow:

1. Open terminal
2. Run some command
3. Select the contents

What if here a lightbulb with sparkle showed like in the editor with Explain This and Add Selection to Chat? Having this would make it possible to ask copilot about a command via only the keyboard:

1. ctrl+shift+up to select above command
2. ctrl+. to open quick fix menu
3. Enter to select first option
Repository: pytorch
Issue: incorrect _unsafe_index meta
Labels: triaged,oncall: pt2,module: fakeTensor,module: decompositions

### 🐛 Describe the bug

```
import torch
import torch._inductor.config

torch.set_default_device("cuda")


inp = torch.empty_strided([4, 512, 96, 96], (1, 4, 196608, 2048), device="cuda")
args = [
    None,
    None,
    torch.empty_strided([192, 1], (1, 1), dtype=torch.int64, device="cuda").zero_(),
    torch.empty_strided([192], (1,), dtype=torch.int64, device="cuda").zero_(),
]

with torch._subclasses.CrossRefFakeMode():
    torch.ops.aten._unsafe_index(inp, args)
```

> RuntimeError: When comparing the output of aten._unsafe_index.Tensor on FakeTensor and concrete Tensors, found mismatched tensor metadata: Stride mismatch! Strides are (36864, 147456, 192, 1) and (18874368, 36864, 192, 1) (mismatched at 0)!


### Versions

master

cc @ezyang @chauhang @penguinwu @SherlockNoMad @zou3519 @bdhirsh @yf225
Repository: pytorch
Issue: BFloat16 support for `torch.fft.rfft`
Labels: triaged,module: complex

### 🚀 The feature, motivation and pitch

I'd like to use `torch.fft.rfft` function with bfloat16 tensor, but the operator doesn't support bfloat16 complex type.
Repro code below:
```python
import torch
x = torch.zeros(3, 4, dtype=torch.bfloat16)
torch.fft.rfft(x)
```
Is there any plan to support this in the near future? Thank you!

### Alternatives

_No response_

### Additional context

_No response_

cc @ezyang @anjali411 @dylanbespalko @mruberry @nikitaved @amjames
Repository: deno
Issue: CJS export analysis should analyze re-exported ES module exports
Labels: bug

Due to require esm, the CJS export anlaysis needs to get smarter and analyze ES module exports.

Version: Deno 2.0.4
Repository: next.js
Issue: Specific data/props case causes infinite loading
Labels: bug,Runtime

### Link to the code that reproduces this issue

https://github.com/chriseling/nextjs-repro-infinite-loading-issue

### To Reproduce

- start the app (yarn dev)
- notice the page shows the loading.tsx and loads infinitely

### Current vs. Expected behavior

expect the client component to load with "hello world"

### Provide environment information

```bash
Operating System:
  Platform: darwin
  Arch: x64
  Version: Darwin Kernel Version 23.6.0: Mon Jul 29 21:13:00 PDT 2024; root:xnu-10063.141.2~1/RELEASE_X86_64
  Available memory (MB): 40960
  Available CPU cores: 16
Binaries:
  Node: 22.0.0
  npm: 10.5.1
  Yarn: 1.22.22
  pnpm: 9.12.2
Relevant Packages:
  next: 15.0.2 // Latest available version is detected (15.0.2).
  eslint-config-next: 15.0.2
  react: 19.0.0-rc-02c0e824-20241028
  react-dom: 19.0.0-rc-02c0e824-20241028
  typescript: 5.6.3
Next.js Config:
  output: N/A
```


### Which area(s) are affected? (Select all that apply)

Runtime

### Which stage(s) are affected? (Select all that apply)

next dev (local), Vercel (Deployed)

### Additional context

_No response_
Repository: svelte
Issue: The documentation section on passing the values layout->page->component and component->page->layout.
Labels: documentation

### Describe the problem

Guys, I would like to have a section in the documentation about transferring values from the layout to the page, to the component and back. Especially when you get used to the new svelte 5 syntax.

### Describe the proposed solution

New section in documentation.

### Importance

would make my life easier
Repository: neovim
Issue: comment: injected filetype is used on linewise motions
Labels: bug,comment

### Problem

See the repro below

### Steps to reproduce

```lua
vim.cmd('echo 1')
return vim.cmd('echo 1')
```
In the above lua file type `gg0fe` then `gcG`. The result is
```lua
"vim.cmd('echo 1')
"return vim.cmd('echo 1')
```


### Expected behavior

```lua
-- vim.cmd('echo 1')
-- return vim.cmd('echo 1')
```
like by `VGgc`.

### Nvim version (nvim -v)

v0.11.0-dev-1067+gb4599acbf

### Vim (not Nvim) behaves the same?

no

### Operating system/version

Debian Sid

### Terminal name/version

alacritty

### $TERM environment variable

alacritty

### Installation

from repo
Repository: godot
Issue: _gui_input is not propagated to parent for keyboard and joy inputs
Labels: enhancement,discussion,topic:input,topic:gui

### Tested versions

- Tested in 4.3.stable

### System information

Godot v4.3.stable - Windows 10.0.22631 - Vulkan (Forward+) - dedicated NVIDIA GeForce GTX 1060 with Max-Q Design (NVIDIA; 31.0.15.5123) - Intel(R) Core(TM) i7-8750H CPU @ 2.20GHz (12 Threads)

### Issue description

_gui_input is only called on the focused control for keyboard and joypad inputs and is not propagated to parent even if accept_event is NOT called.
Note that is does not affect mouse events (when the mouse_filter of the child is set to Pass).

### Steps to reproduce

- Launch the MRP, the button on screen will be automatically focused
- Press space, notice in the output that _gui_input is only called on the child.
- Notice that if you move the mouse on the button or even click on it, _gui_input is called on both the child and the parent.

### Minimal reproduction project (MRP)

[bug-gui-input-propagation.zip](https://github.com/user-attachments/files/17579412/bug-gui-input-propagation.zip)
Repository: flutter
Issue: ShaderCompilerException: Could not write file to build\flutter_assets\shaders/ink_sparkle.frag
Labels: c: crash,P2,team-tool,triaged-tool

First known to have appeared (at least with this stack trace) in 3.23.0

```
ShaderCompilerException: Shader compilation of "C:\Development\flutter\packages\flutter\lib\src\material\shaders\ink_sparkle.frag" to "build\flutter_assets\shaders/ink_sparkle.frag" failed with exit code 1.
impellerc stdout:
impellerc stderr:
Could not write file to build\flutter_assets\shaders/ink_sparkle.frag

at ShaderCompiler.compileShader(shader_compiler.dart:190)
at <asynchronous gap>(async)
at writeBundle.<anonymous closure>(bundle_builder.dart:221)
at <asynchronous gap>(async)
at Future.wait.<anonymous closure>(future.dart:534)
at <asynchronous gap>(async)
at writeBundle(bundle_builder.dart:187)
at <asynchronous gap>(async)
at WebDevFS.update(devfs_web.dart:1005)
at <asynchronous gap>(async)
at ResidentWebRunner._updateDevFS(resident_web_runner.dart:583)
at <asynchronous gap>(async)
at ResidentWebRunner.run.<anonymous closure>(resident_web_runner.dart:334)
at <asynchronous gap>(async)
at asyncGuard.<anonymous closure>(async_guard.dart:111)
at <asynchronous gap>(async)
```

source:

https://github.com/flutter/flutter/blob/ac7879e2aa6de40afec1fe2af9730a8d55de3e06/packages/flutter_tools/lib/src/build_system/tools/shader_compiler.dart#L189-L196
Repository: go
Issue: os: TestPipeThreads failing sporadically on aix-ppc64
Labels: NeedsInvestigation

### Go version

master

### Output of `go env` in your module/workspace:

```shell
GOOS=aix
```


### What did you do?

```
#build go.
go test os -test.run=TestPipeThreads -test.count=1
```

### What did you see happen?

The tests fails consistently on aix systems with GOMAXPROC >= 8. It fails sporadically with smaller values, as is seen with aix-ppc64 CI. This CI [log](https://build.golang.org/log/f58a9df223e4f6179ce1ec967c5ec8fa219cdefa) seems like the first of the recent failures. However, it is reproducible prior to the 1.23 release.

### What did you expect to see?

The test passes consistently.
Repository: vscode
Issue: Typo in debugger
Labels: under-discussion


Type: <b>Bug</b>

There's a typo in debugger in the breakpoints. Specifically, "Toggle Activate Breakpoints" should probably be "Toggle active breakpoints".

VS Code version: Code 1.92.0 (Universal) (b1c0a14de1414fcdaa400695b4db1c0799bc3124, 2024-07-31T23:26:45.634Z)
OS version: Darwin arm64 24.0.0
Modes:

<details>
<summary>System Info</summary>

|Item|Value|
|---|---|
|CPUs|Apple M1 Max (10 x 2400)|
|GPU Status|2d_canvas: enabled<br>canvas_oop_rasterization: enabled_on<br>direct_rendering_display_compositor: disabled_off_ok<br>gpu_compositing: enabled<br>multiple_raster_threads: enabled_on<br>opengl: enabled_on<br>rasterization: enabled<br>raw_draw: disabled_off_ok<br>skia_graphite: disabled_off<br>video_decode: enabled<br>video_encode: enabled<br>webgl: enabled<br>webgl2: enabled<br>webgpu: enabled<br>webnn: disabled_off|
|Load (avg)|4, 3, 3|
|Memory (System)|64.00GB (3.20GB free)|
|Process Argv|--crash-reporter-id eb3ac444-ca87-45c1-b2ff-c5ac56c7837e|
|Screen Reader|no|
|VM|0%|
</details><details><summary>Extensions (35)</summary>

Extension|Author (truncated)|Version
---|---|---
c3-ai-dx-v8|c3a|4.0.6
vale-vscode|chr|0.20.0
excel-to-markdown-table|csh|1.3.0
vscode-markdownlint|Dav|0.56.0
EditorConfig|Edi|0.16.4
vscode-pull-request-github|Git|0.98.0
gc-excelviewer|Gra|4.2.62
rest-client|hum|0.25.1
markdown-shortcuts|mdi|0.12.0
rainbow-csv|mec|3.12.0
debugpy|ms-|2024.12.0
isort|ms-|2023.10.1
python|ms-|2024.14.1
vscode-pylance|ms-|2024.10.1
jupyter|ms-|2024.7.0
jupyter-keymap|ms-|1.1.2
jupyter-renderers|ms-|1.0.19
vscode-jupyter-cell-tags|ms-|0.1.9
vscode-jupyter-slideshow|ms-|0.1.6
remote-containers|ms-|0.388.0
fabric8-analytics|red|0.9.5
java|red|1.35.1
vscode-xml|red|0.27.1
code-spell-checker|str|3.0.1
markdowntable|Tak|0.12.0
intellicode-api-usage-examples|Vis|0.2.8
vscode-gradle|vsc|3.16.4
vscode-java-debug|vsc|0.58.0
vscode-java-dependency|vsc|0.24.0
vscode-java-pack|vsc|0.29.0
vscode-java-test|vsc|0.42.0
vscode-maven|vsc|0.44.0
markdown-pdf|yza|1.5.0
markdown-all-in-one|yzh|3.6.2
grammarly|znc|0.24.0


</details><details>
<summary>A/B Experiments</summary>

```
vsliv368:30146709
vspor879:30202332
vspor708:30202333
vspor363:30204092
vscod805:30301674
binariesv615:30325510
vsaa593:30376534
py29gd2263:31024239
c4g48928:30535728
azure-dev_surveyone:30548225
962ge761:30959799
pythongtdpath:30769146
pythonnoceb:30805159
asynctok:30898717
pythonmypyd1:30879173
h48ei257:31000450
pythontbext0:30879054
cppperfnew:31000557
dsvsc020:30976470
pythonait:31006305
dsvsc021:30996838
945dj816:31013170
dvdeprecation:31068756
dwnewjupytercf:31046870
impr_priority:31102340
nativerepl2:31139839
refactort:31108082
pythonrstrctxt:31112756
wkspc-onlycs-t:31132770
wkspc-ranged-t:31151552
iacca2:31156134
notype1cf:31157160
5fd0e150:31155592

```

</details>

<!-- generated by issue reporter -->
Repository: pytorch
Issue: [FR] Support sub-group partition ProcessGroup
Labels: oncall: distributed,triaged

### 🚀 The feature, motivation and pitch

This is something I'd like to achieve:

```py
def custom_partition_data_and_model_subgroup(input_group: ProcessGroup):
     node_list = get_node_list(input_group)
     outer_process_group , inner_process_group = partition_subgroup(node_list, subgroup_size=4)
     return outer_process_group , inner_process_group
```

Is it something that can be achieved by current Pytorch distributed interface (lacking doc)? or something that can be supported in the future?

### Alternatives

_No response_

### Additional context

_No response_

cc @H-Huang @awgu @kwen2501 @wanchaol @fegin @fduwjj @wz337 @wconstab @d4l3k @c-p-i-o
Repository: vscode
Issue: Narrow sidebar chat layout issues
Labels: bug,panel-chat

Continue https://github.com/microsoft/vscode/issues/231369

- Padding around welcome view text
- And wrapped edits title:

![Image](https://github.com/user-attachments/assets/4df94ba4-c0b3-4bf7-a7b7-2ebe42371506)
Repository: pytorch
Issue: Enable opting out of CI experiments
Labels: triaged

We use this issue to configure issue rollouts and allow people to opt into experiments early.
https://github.com/pytorch/test-infra/issues/5132

However, there's no mechanism to opt out if desired.

Ask:
Update target_determinator.py (which parses the above issue) to allow people to opt out of a given experiment by prefixing the experiment with a `-`.  

For example, specifying `@SomeUser,lf,-awsa100` would opt SomeUser into the `lf` experiment but out of the `awsa100` experiment.
Repository: langchain
Issue: ImportError: Dependencies for InstructorEmbedding not found.
Labels: 🤖:bug

### Checked other resources

- [X] I added a very descriptive title to this issue.
- [X] I searched the LangChain documentation with the integrated search.
- [X] I used the GitHub search to find a similar question and didn't find it.
- [X] I am sure that this is a bug in LangChain rather than my code.
- [X] The bug is not resolved by updating to the latest stable version of LangChain (or the specific integration package).

### Example Code

```python

from langchain_community.embeddings import HuggingFaceInstructEmbeddings

DEVICE = "cpu"

#Initialize embeddings using a pre-trained model to represent the text data.
embedddings_model = "sentence-transformers/multi-qa-distilbert-cos-v1"

embeddings = HuggingFaceInstructEmbeddings(
    model_name=embedddings_model,
    model_kwargs={"device": DEVICE}
)
```

### Error Message and Stack Trace (if applicable)

```
USER_AGENT environment variable not set, consider setting it to identify your requests.
Traceback (most recent call last):
  File "/home/tda/miniconda3/envs/ragbot2/lib/python3.10/site-packages/langchain_community/embeddings/huggingface.py", line 188, in __init__
    from InstructorEmbedding import INSTRUCTOR
ModuleNotFoundError: No module named 'InstructorEmbedding'

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/home/tda/GenAICourse/RAG/ragbot_langchain/app.py", line 38, in <module>
    embeddings = HuggingFaceInstructEmbeddings(
  File "/home/tda/miniconda3/envs/ragbot2/lib/python3.10/site-packages/langchain_community/embeddings/huggingface.py", line 194, in __init__
    raise ImportError("Dependencies for InstructorEmbedding not found.") from e
ImportError: Dependencies for InstructorEmbedding not found.
```

### Description

I'm trying to build embeddings for a RAG application and I get the error 
ImportError: Dependencies for InstructorEmbedding not found.

### System Info

System Information
------------------
> OS:  Linux
> OS Version:  #134~20.04.1-Ubuntu SMP Tue Oct 1 15:27:33 UTC 2024
> Python Version:  3.10.0 (default, Mar  3 2022, 09:58:08) [GCC 7.5.0]

Package Information
-------------------
> langchain_core: 0.3.14
> langchain: 0.3.5
> langchain_community: 0.3.3
> langsmith: 0.1.138
> langchain_chroma: 0.1.4
> langchain_mistralai: 0.2.0
> langchain_openai: 0.2.4
> langchain_text_splitters: 0.3.1

Optional packages not installed
-------------------------------
> langgraph
> langserve

Other Dependencies
------------------
> aiohttp: 3.10.10
> async-timeout: 4.0.3
> chromadb: 0.5.16
> dataclasses-json: 0.6.7
> fastapi: 0.115.4
> httpx: 0.27.2
> httpx-sse: 0.4.0
> jsonpatch: 1.33
> numpy: 1.26.4
> openai: 1.53.0
> orjson: 3.10.10
> packaging: 24.1
> pydantic: 2.9.2
> pydantic-settings: 2.6.0
> PyYAML: 6.0.2
> requests: 2.32.3
> requests-toolbelt: 1.0.0
> SQLAlchemy: 2.0.36
> tenacity: 9.0.0
> tiktoken: 0.8.0
> tokenizers: 0.20.1
> typing-extensions: 4.12.2
Repository: rust
Issue: rust-gdb: Accessing tuple inside enum triggers invalid gdb expression syntax error.
Labels: C-bug,WG-debugging

<!--
Thank you for filing a bug report! 🐛 Please provide a short summary of the bug,
along with any information you feel relevant to replicating the bug.
-->

I tried this code:

```rust
fn main() {
    let a = Some((0u32, 1u32));
    dbg!(a);
}
```

When debugging in gdb, I expect to be able to write an expression to access a value inside a tuple, inside an enum. I want to do that so that I can e.g. dereference a pointer stored in such a tuple. It seems that `a.0` accesses the `Some` branch of my option, so I would expect `a.0.0` to access the first tuple elem.

Instead, `a.0.0` triggers a gdb expression syntax error. Note that it *does* work to append `.0` to a gdb variable storing `a.0` to access the tuple, which further suggests that `a.0.0` should work.

```
Breakpoint 2.1, gdb_tuple::main () at src/main.rs:3
3	    dbg!(a);
(gdb) p a
$1 = core::option::Option<(u32, u32)>::Some((
    0,
    1
  ))
(gdb) p a.0
$2 = (
  0,
  1
)
(gdb) p a.0.0
field name expected
(gdb) p $2.0
$3 = 0
```

`a.0` to access the `Some` admittedly seems like a bit of black magic. Helper functions like `unwrap` are consistently not available in the binaries that I've tried to debug thus far, so I'm not sure what else I should be doing.

### Meta
<!--
If you're using the stable version of the compiler, you should also check if the
bug also exists in the beta or nightly versions.
-->

`rustc --version --verbose`:
```
rustc 1.84.0-nightly (439284741 2024-10-21)
binary: rustc
commit-hash: 4392847410ddd67f6734dd9845f9742ff9e85c83
commit-date: 2024-10-21
host: x86_64-unknown-linux-gnu
release: 1.84.0-nightly
LLVM version: 19.1.1
```
Repository: pytorch
Issue: [ROCm] sdpa group query attention bf16 numeric error
Labels: module: rocm,triaged

### 🐛 Describe the bug

Hi AMD Team,

On MI300X pytorch nightly grouped query attention is running into numeric errors. I have confirmed on H100 that this script does not have numeric errors.

Can you look into this & potentially add an numeric unit test for this?

cc: @hliuca

## ROCm Error 
```python
    raise error_metas[0].to_error(msg)
AssertionError: Tensor-likes are not close!

Mismatched elements: 8388584 / 8388608 (100.0%)
Greatest absolute difference: 0.99609375 at index (0, 0, 0, 15) (up to 1e-05 allowed)
Greatest relative difference: inf at index (0, 0, 1, 0) (up to 0.016 allowed)
```

## Reprod Script
```python
import torch
from torch.nn.functional import scaled_dot_product_attention

batch = 4
seq_len_q = 1024
seq_len_kv = 1024
D = 128

query = torch.randn(batch, 32, seq_len_q, D, device='cuda', dtype=torch.bfloat16)
key = torch.randn(batch, 8, seq_len_kv, D, device='cuda', dtype=torch.bfloat16)
value = torch.randn(batch, 8, seq_len_kv, D, device='cuda', dtype=torch.bfloat16)


output_gqa = scaled_dot_product_attention(query, key, value, is_causal=True, enable_gqa=True)

key = key.repeat_interleave(4,1)
value = value.repeat_interleave(4,1)
output_repeat = scaled_dot_product_attention(query, key, value, is_causal=True)

torch.testing.assert_close(output_gqa, output_repeat)
```

### Versions

## ROCm Versions

```
~$ pip list | grep torch
pytorch-triton-rocm 3.1.0+cf34004b8a
torch               2.6.0.dev20241030+rocm6.2
```

## H100 Versions

```
~$ pip list | grep torch
pytorch-triton               3.1.0+cf34004b8a
torch                        2.6.0.dev20241030+cu124
```

cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang @naromero77amd