46caad52ac
If a `do concurrent` loop is offloaded then there should be no CUDA data transfer in it. Update the semantic and lowering to take that into account. `AssignmentChecker` has to be put into a separate pass because the checkers in `SemanticsVisitor` cannot have the same `Enter/Leave` functions. The `DoForallChecker` already has `Eneter/Leave` functions for the `DoConstruct`.
63 lines
1.9 KiB
C++
63 lines
1.9 KiB
C++
//===-- lib/Semantics/check-cuda.h ------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef FORTRAN_SEMANTICS_CHECK_CUDA_H_
|
|
#define FORTRAN_SEMANTICS_CHECK_CUDA_H_
|
|
|
|
#include "flang/Semantics/semantics.h"
|
|
#include <list>
|
|
|
|
namespace Fortran::parser {
|
|
struct Program;
|
|
class Messages;
|
|
struct Name;
|
|
class CharBlock;
|
|
struct AssignmentStmt;
|
|
struct ExecutionPartConstruct;
|
|
struct ExecutableConstruct;
|
|
struct ActionStmt;
|
|
struct IfConstruct;
|
|
struct CUFKernelDoConstruct;
|
|
struct SubroutineSubprogram;
|
|
struct FunctionSubprogram;
|
|
struct SeparateModuleSubprogram;
|
|
} // namespace Fortran::parser
|
|
|
|
namespace Fortran::semantics {
|
|
|
|
class SemanticsContext;
|
|
|
|
class CUDAChecker : public virtual BaseChecker {
|
|
public:
|
|
explicit CUDAChecker(SemanticsContext &c) : context_{c} {}
|
|
void Enter(const parser::SubroutineSubprogram &);
|
|
void Enter(const parser::FunctionSubprogram &);
|
|
void Enter(const parser::SeparateModuleSubprogram &);
|
|
void Enter(const parser::CUFKernelDoConstruct &);
|
|
void Leave(const parser::CUFKernelDoConstruct &);
|
|
void Enter(const parser::AssignmentStmt &);
|
|
void Enter(const parser::OpenACCBlockConstruct &);
|
|
void Leave(const parser::OpenACCBlockConstruct &);
|
|
void Enter(const parser::OpenACCCombinedConstruct &);
|
|
void Leave(const parser::OpenACCCombinedConstruct &);
|
|
void Enter(const parser::OpenACCLoopConstruct &);
|
|
void Leave(const parser::OpenACCLoopConstruct &);
|
|
void Enter(const parser::DoConstruct &);
|
|
void Leave(const parser::DoConstruct &);
|
|
|
|
private:
|
|
SemanticsContext &context_;
|
|
int deviceConstructDepth_{0};
|
|
};
|
|
|
|
bool CanonicalizeCUDA(parser::Program &);
|
|
|
|
} // namespace Fortran::semantics
|
|
|
|
#endif // FORTRAN_SEMANTICS_CHECK_CUDA_H_
|