Home
last modified time | relevance | path

Searched refs:NumTeams (Results 1 – 24 of 24) sorted by relevance

/llvm-project-15.0.7/openmp/libomptarget/src/
H A DLegacyAPI.cpp118 void **ArgMappers, int32_t NumTeams, in __tgt_target_teams_mapper() argument
124 return __tgt_target_kernel(Loc, DeviceId, NumTeams, ThreadLimit, HostPtr, in __tgt_target_teams_mapper()
130 int64_t *ArgTypes, int32_t NumTeams, in __tgt_target_teams() argument
135 NumTeams, ThreadLimit); in __tgt_target_teams()
141 int64_t *ArgTypes, int32_t NumTeams, in __tgt_target_teams_nowait() argument
149 NumTeams, ThreadLimit); in __tgt_target_teams_nowait()
155 map_var_info_t *ArgNames, void **ArgMappers, int32_t NumTeams, in __tgt_target_teams_nowait_mapper() argument
162 ArgMappers, NumTeams, ThreadLimit); in __tgt_target_teams_nowait_mapper()
H A Dinterface.cpp211 EXTERN int __tgt_target_kernel(ident_t *Loc, int64_t DeviceId, int32_t NumTeams, in __tgt_target_kernel() argument
241 bool IsTeams = NumTeams != -1; in __tgt_target_kernel()
243 NumTeams = 0; in __tgt_target_kernel()
249 Args->ArgMappers, NumTeams, ThreadLimit, Args->Tripcount, in __tgt_target_kernel()
259 ident_t *Loc, int64_t DeviceId, int32_t NumTeams, int32_t ThreadLimit, in __tgt_target_kernel_nowait() argument
264 return __tgt_target_kernel(Loc, DeviceId, NumTeams, ThreadLimit, HostPtr, in __tgt_target_kernel_nowait()
H A Ddevice.cpp601 int32_t NumTeams, int32_t ThreadLimit, in runTeamRegion() argument
606 TgtOffsets, TgtVarsSize, NumTeams, ThreadLimit, in runTeamRegion()
609 TgtOffsets, TgtVarsSize, NumTeams, in runTeamRegion()
/llvm-project-15.0.7/openmp/libomptarget/plugins/amdgpu/src/
H A Drtl.cpp348 int NumTeams; member
1279 if (NumTeams > 0) { in getLaunchVals()
1280 NumGroups = (NumTeams < NumGroups) ? NumTeams : NumGroups; in getLaunchVals()
1288 if (Env.NumTeams > 0) { in getLaunchVals()
1289 NumGroups = (Env.NumTeams < NumGroups) ? Env.NumTeams : NumGroups; in getLaunchVals()
1295 if (NumTeams <= 0) { in getLaunchVals()
1321 if (NumGroups > NumTeams && NumTeams > 0) { in getLaunchVals()
1325 NumTeams); in getLaunchVals()
1330 if (NumTeams > 0) { in getLaunchVals()
1331 NumGroups = NumTeams; in getLaunchVals()
[all …]
/llvm-project-15.0.7/openmp/libomptarget/DeviceRTL/src/
H A DReduction.cpp205 uint32_t NumTeams = omp_get_num_teams(); in __kmpc_nvptx_teams_reduce_nowait_v2() local
260 unsigned NumRecs = kmpcMin(NumTeams, uint32_t(num_of_records)); in __kmpc_nvptx_teams_reduce_nowait_v2()
261 if (ChunkTeamCount == NumTeams - Bound - 1) { in __kmpc_nvptx_teams_reduce_nowait_v2()
/llvm-project-15.0.7/openmp/libomptarget/include/
H A Domptargetplugin.h145 int32_t NumTeams, int32_t ThreadLimit,
151 int32_t NumTeams, int32_t ThreadLimit, uint64_t LoopTripcount,
H A Domptarget.h335 int __tgt_target_kernel(ident_t *Loc, int64_t DeviceId, int32_t NumTeams,
338 int __tgt_target_kernel_nowait(ident_t *Loc, int64_t DeviceId, int32_t NumTeams,
H A Ddevice.h431 int32_t NumTeams, int32_t ThreadLimit,
/llvm-project-15.0.7/openmp/libomptarget/plugins/cuda/src/
H A Drtl.cpp166 int NumTeams = 0; member
743 DeviceData[DeviceId].NumTeams = EnvNumTeams; in initDevice()
745 DeviceData[DeviceId].NumTeams = DeviceRTLTy::DefaultNumTeams; in initDevice()
750 if (DeviceData[DeviceId].NumTeams > DeviceData[DeviceId].BlocksPerGrid) { in initDevice()
753 DeviceData[DeviceId].NumTeams = DeviceData[DeviceId].BlocksPerGrid; in initDevice()
1211 DP("Using default number of teams %d\n", DeviceData[DeviceId].NumTeams); in runTargetTeamRegion()
1212 CudaBlocksPerGrid = DeviceData[DeviceId].NumTeams; in runTargetTeamRegion()
/llvm-project-15.0.7/openmp/libomptarget/DeviceRTL/include/
H A DInterface.h294 void __kmpc_push_num_teams(IdentTy *Loc, int32_t TId, int32_t NumTeams,
/llvm-project-15.0.7/clang/lib/CodeGen/
H A DCGOpenMPRuntimeGPU.h205 void emitNumTeamsClause(CodeGenFunction &CGF, const Expr *NumTeams,
H A DCGOpenMPRuntime.cpp6574 const Expr *NumTeams = in getNumTeamsExprForTargetDirective() local
6576 if (NumTeams->isIntegerConstantExpr(CGF.getContext())) in getNumTeamsExprForTargetDirective()
6580 return NumTeams; in getNumTeamsExprForTargetDirective()
6603 const Expr *NumTeams = in getNumTeamsExprForTargetDirective() local
6605 if (NumTeams->isIntegerConstantExpr(CGF.getContext())) in getNumTeamsExprForTargetDirective()
6608 return NumTeams; in getNumTeamsExprForTargetDirective()
6693 if (NumTeams != nullptr) { in emitNumTeamsForTargetDirective()
6701 llvm::Value *NumTeamsVal = CGF.EmitScalarExpr(NumTeams, in emitNumTeamsForTargetDirective()
6712 llvm::Value *NumTeamsVal = CGF.EmitScalarExpr(NumTeams, in emitNumTeamsForTargetDirective()
10998 const Expr *NumTeams, in emitNumTeamsClause() argument
[all …]
H A DCGOpenMPRuntime.h1652 virtual void emitNumTeamsClause(CodeGenFunction &CGF, const Expr *NumTeams,
2476 void emitNumTeamsClause(CodeGenFunction &CGF, const Expr *NumTeams,
H A DCGOpenMPRuntimeGPU.cpp1236 const Expr *NumTeams, in emitNumTeamsClause() argument
H A DCGStmtOpenMP.cpp6688 const Expr *NumTeams = NT ? NT->getNumTeams() : nullptr; in emitCommonOMPTeamsDirective() local
6691 CGF.CGM.getOpenMPRuntime().emitNumTeamsClause(CGF, NumTeams, ThreadLimit, in emitCommonOMPTeamsDirective()
/llvm-project-15.0.7/openmp/libomptarget/plugins/ve/src/
H A Drtl.cpp409 int32_t NumTeams, int32_t ThreadLimit, in __tgt_rtl_run_target_team_region() argument
/llvm-project-15.0.7/clang/include/clang/AST/
H A DOpenMPClause.h5973 Stmt *NumTeams = nullptr; variable
5978 void setNumTeams(Expr *E) { NumTeams = E; } in setNumTeams()
5994 OMPClauseWithPreInit(this), LParenLoc(LParenLoc), NumTeams(E) { in OMPNumTeamsClause()
6011 Expr *getNumTeams() { return cast<Expr>(NumTeams); } in getNumTeams()
6014 Expr *getNumTeams() const { return cast<Expr>(NumTeams); } in getNumTeams()
6016 child_range children() { return child_range(&NumTeams, &NumTeams + 1); } in children()
6019 return const_child_range(&NumTeams, &NumTeams + 1); in children()
/llvm-project-15.0.7/llvm/include/llvm/Frontend/OpenMP/
H A DOMPIRBuilder.h845 Value *Ident, Value *DeviceID, Value *NumTeams,
/llvm-project-15.0.7/flang/lib/Parser/
H A Dopenmp-parsers.cpp235 "NUM_TEAMS" >> construct<OmpClause>(construct<OmpClause::NumTeams>(
/llvm-project-15.0.7/llvm/lib/Frontend/OpenMP/
H A DOMPIRBuilder.cpp796 Value *DeviceID, Value *NumTeams, Value *NumThreads, Value *HostPtr, in emitTargetKernel() argument
812 SmallVector<Value *> OffloadingArgs{Ident, DeviceID, NumTeams, in emitTargetKernel()
/llvm-project-15.0.7/flang/lib/Semantics/
H A Dcheck-omp-structure.cpp1873 CHECK_REQ_SCALAR_INT_CLAUSE(NumTeams, OMPC_num_teams) in CHECK_SIMPLE_CLAUSE()
/llvm-project-15.0.7/clang/lib/Sema/
H A DTreeTransform.h1978 OMPClause *RebuildOMPNumTeamsClause(Expr *NumTeams, SourceLocation StartLoc, in RebuildOMPNumTeamsClause() argument
1981 return getSema().ActOnOpenMPNumTeamsClause(NumTeams, StartLoc, LParenLoc, in RebuildOMPNumTeamsClause()
H A DSemaOpenMP.cpp22167 OMPClause *Sema::ActOnOpenMPNumTeamsClause(Expr *NumTeams, in ActOnOpenMPNumTeamsClause() argument
22171 Expr *ValExpr = NumTeams; in ActOnOpenMPNumTeamsClause()
/llvm-project-15.0.7/clang/include/clang/Sema/
H A DSema.h11747 OMPClause *ActOnOpenMPNumTeamsClause(Expr *NumTeams, SourceLocation StartLoc,