Please make sure that this is a bug. As per our
GitHub Policy,
we only address code/doc bugs, performance issues, feature requests and
build/installation issues on GitHub. tag:bug_template

System information

• Have I written custom code (as opposed to using a stock
  example script provided in TensorFlow): Yes

🐛 Bug

Follow up on #43476.
Scanning CUDA KernelUtils.GET_BLOCKS might be called with overflowed 32bit integers. should convert to int64 to hold num_kernels

Additional context

See examples such as:
https://github.com/pytorch/pytorch/blob/c010ef7f0c6d837809a7e973048afac76373e3de/aten/src/ATen/native/cuda/vol2col.cuh#L107

https://github.com/pytorch/pytorch/blob/c010ef7f0c6d83780

Reference from TensorFlow: https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/matrix-band-part

This op is used by the Music Transformer model.

We plan to gradually migrate brain.js to TypeScript, code base is pretty large, so we would love your help! 💪

How to contribute?

• Convert a file from .js to .ts
• Add types, fix all type errors.
• Submit a PR! 🎉

Here you can find a guide on how to contribute.

Want to convert something, let us know in the comment an

Bug Report

These tests were run on s390x. s390x is big-endian architecture.

Failure log for helper_test.py

________________________________________________ TestHelperTensorFunctions.test_make_tensor ________________________________________________

self = <helper_test.TestHelperTensorFunctions testMethod=test_make_tensor>

    def test_make_tensor(self):  # type: () -> None
    

What would you like to be added: As title

Why is this needed: All pruning schedule except AGPPruner only support level, L1, L2. While there are FPGM, APoZ, MeanActivation and Taylor, it would be much better if we can choose any pruner with any pruning schedule.

**Without this feature, how does current nni

Hi I would like to propose a better implementation for 'test_indices':

We can remove the unneeded np.array casting:

Cleaner/New:
test_indices = list(set(range(len(texts))) - set(train_indices))

Old:
test_indices = np.array(list(set(range(len(texts))) - set(train_indices)))