Int based RNG

[csarofeen]

No description provided.

[github-actions]

PR Reviewer Guide 🔍

(Review updated until commit 2fa293d)

Here are some key observations to aid the review process:

⏱️ Estimated effort to review: 4 🔵🔵🔵🔵⚪

🧪 PR contains tests

⚡ Recommended focus areas for review Logic Change The condition in the if statement has been modified to also check if tv->getMemoryType() == MemoryType::Local. This change may affect the behavior of the function. if (tv && (aligned_array_of_regs_.count(tv) || tv->getMemoryType() == MemoryType::Local)) { New Functionality A new function addRNG has been added, which appears to insert RNG (Random Number Generator) operations into the expression list. This new functionality may require additional testing and verification. std::vector<Expr*> addRNG(const std::vector<Expr*>& exprs) { FUSER_PERF_SCOPE("GpuLower::Lower::addRNG"); // Check if magic zero was even used, if not we don't have to define it or // update it. const auto gpu_lower = GpuLower::current(); auto kernel = gpu_lower->kernel(); const bool has_rng = std::any_of( kernel->exprs().begin(), kernel->exprs().end(), [](Expr* expr) { return expr->isA<RNGOp>(); }); if (!has_rng) { return exprs; } auto exprs_ = RNGInserter::insert(exprs); std::cout << "====================" << std::endl; for (auto expr : exprs_) { std::cout << expr->toString() << std::endl; } std::cout << "====================" << std::endl; // NVF_THROW("throw"); return exprs_; } Enum Modification A new value Philox has been added to the TernaryOpType enum. This change may affect the behavior of the program and requires verification. enum class TernaryOpType { Clamp, Lerp, Threshold, Where, Philox };