Search results

The substantial overhead of NKI kernel construction in the Python call stack.

bugSomething isn't workingSomething isn't working

Status: Open.

#72 In aws-neuron/nki-samples;

· dinghongsong opened on May 1, 2025

Wrong instruction dependency

bugSomething isn't workingSomething isn't working

Status: Open.

#69 In aws-neuron/nki-samples;

· jinhongyii opened on Apr 14, 2025

[Feature Request] Fine-grained communication API

feature-request

Status: Open.

#65 In aws-neuron/nki-samples;

· jinhongyii opened on Apr 4, 2025

Some APIs are not documented including tiled_offloaded_* and nccl ops

documentationImprovements or additions to documentationImprovements or additions to documentation

Status: Open.

#60 In aws-neuron/nki-samples;

· akothen opened on Mar 24, 2025

The nisa.nc_matmul in NKI is not as well-optimized as torch.matmul.

bugSomething isn't workingSomething isn't working

Status: Open.

#59 In aws-neuron/nki-samples;

· dinghongsong opened on Mar 23, 2025

Flash Attention Integration Tests Fail

bugSomething isn't workingSomething isn't working

Status: Open.

#52 In aws-neuron/nki-samples;

· weiT1993 opened on Feb 19, 2025

Fused Self Attention Produces Large Errors When Using torch.randn Instead of torch.rand

bugSomething isn't workingSomething isn't working

Status: Open.

#51 In aws-neuron/nki-samples;

· Jerry2423 opened on Feb 18, 2025