NameError: name '_seed' is not defined

[TED-EE]

Reproduce steps:

git clone https://github.com/FlagOpen/FlagGems.git
cd FlagGems
pip install .

cd tests
pytest test_xx_ops.py

3. error:

tests/test_binary_pointwise_ops.py:3: in <module>
    import flag_gems
/usr/local/lib/python3.10/dist-packages/flag_gems/__init__.py:4: in <module>
    from .ops import *  # noqa: F403
/usr/local/lib/python3.10/dist-packages/flag_gems/ops/__init__.py:21: in <module>
    from .dropout import native_dropout
/usr/local/lib/python3.10/dist-packages/flag_gems/ops/dropout.py:24: in <module>
    del _seed
E   NameError: name '_seed' is not defined

Solution:
I tried to workaround this problem by defining _seed and _offset, hacking FlagGems/src/flag_gems/ops/dropout.py from

try:
    tl_rand_dtype = tl.int64

    @triton.jit
    def _rand(seed, offset):
        offset = offset.to(tl_rand_dtype)

    _grid = (1,)
    _seed, _offset = philox_cuda_seed_offset(0)
    _rand[_grid](_seed, _offset)
except Exception:
    tl_rand_dtype = tl.int32

to

try:
    tl_rand_dtype = tl.int64

    @triton.jit
    def _rand(seed, offset):
        offset = offset.to(tl_rand_dtype)

    _seed = 0
    _offset = 0
    _grid = (1,)
    _seed, _offset = philox_cuda_seed_offset(0)
    _rand[_grid](_seed, _offset)
except Exception:
    tl_rand_dtype = tl.int32

, and it worked properly.

Is there any formal solution for it?

[StrongSpoon]

operator dropout has been updated. please take a try on the new commit.

[TED-EE]

It works. Thanks.