Files
mscclpp/python/test/fifo_test.cu
Binyang2014 097aa8843a Fix pytest unstable issue. (#170)
- remove `#include <cstdint>` from `poll.hpp`. To make it only contains
device-side code
- Fix compilation issue, which will cause pytest fail randomly. Reuse
the compiled result for same kernel with different arguments
2023-09-06 17:09:04 -07:00

13 lines
322 B
Plaintext

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.
#include "common.hpp"
// other headers
#include "mscclpp/fifo_device.hpp"
extern "C" __global__ void __launch_bounds__(1024, 1) fifo(mscclpp::FifoDeviceHandle fifo) {
mscclpp::ProxyTrigger trigger;
trigger.fst = 123;
fifo.push(trigger);
}