Animesh Jain comments

Results 136 comments of


                                            Animesh Jain

Reproducing NNPACK numbers on SKL i5-6600K

How many threads are running here? Can you control the number of threads for NNPACK. Even when I am setting OMP_NUM_THREADS to 1, I can see multiple threads running in...

Support static_argnums for kwargs

Thanks @hyunwoongko for filing the issue. We also came across this type of issue quite recently. I have assigned it to myself.

Check Grad state before backprop AND preserve RNG state

> Can we just check if grad is enabled before we call autograd.grad? So silly of me. Yeah, fixed this. > I do wonder whether we can have a more...

[dynamo][guards-cpp-refactor] NO_TENSOR_ALIASING guard

@jansel Ready for another review.

[inductor] Check memory compression ratio in model tests

Compression ratio seems pretty bad for small batch sizes. Maybe it's because of fallback. Cc @ngimel

[dynamo][source] Remove inspect getattr_static from AttrSource

@pytorchbot merge

[dynamo][prepare for nn module guards] Guard nn modules for a few benchmarks

@pytorchbot merge

[dynamo] use lazy disable dynamo for manual seed

@pytorchbot merge

[dynamo] use lazy disable dynamo for manual seed

@pytorchbot merge

[dynamo] use lazy disable dynamo for manual seed

@pytorchbot merge -f "stuck ci"