File tree Expand file tree Collapse file tree 4 files changed +8
-8
lines changed Expand file tree Collapse file tree 4 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -573,8 +573,8 @@ struct AMDGPUKernelTy : public GenericKernelTy {
573573 // / Return maximum block size for maximum occupancy
574574 // /
575575 // / TODO: This needs to be implemented for amdgpu
576- Expected<size_t > maxGroupSize (GenericDeviceTy &GenericDevice,
577- size_t DynamicMemSize) const override {
576+ Expected<uint64_t > maxGroupSize (GenericDeviceTy &GenericDevice,
577+ uint64_t DynamicMemSize) const override {
578578 return 1 ;
579579 }
580580
Original file line number Diff line number Diff line change @@ -388,8 +388,8 @@ struct GenericKernelTy {
388388 KernelLaunchParamsTy LaunchParams,
389389 AsyncInfoWrapperTy &AsyncInfoWrapper) const = 0;
390390
391- virtual Expected<size_t > maxGroupSize (GenericDeviceTy &GenericDevice,
392- size_t DynamicMemSize) const = 0;
391+ virtual Expected<uint64_t > maxGroupSize (GenericDeviceTy &GenericDevice,
392+ uint64_t DynamicMemSize) const = 0;
393393
394394 // / Get the kernel name.
395395 const char *getName () const { return Name.c_str (); }
Original file line number Diff line number Diff line change @@ -158,8 +158,8 @@ struct CUDAKernelTy : public GenericKernelTy {
158158 AsyncInfoWrapperTy &AsyncInfoWrapper) const override ;
159159
160160 // / Return maximum block size for maximum occupancy
161- Expected<size_t > maxGroupSize (GenericDeviceTy &,
162- size_t DynamicMemSize) const override {
161+ Expected<uint64_t > maxGroupSize (GenericDeviceTy &,
162+ uint64_t DynamicMemSize) const override {
163163 int minGridSize;
164164 int maxBlockSize;
165165 auto Res = cuOccupancyMaxPotentialBlockSize (
Original file line number Diff line number Diff line change @@ -115,8 +115,8 @@ struct GenELF64KernelTy : public GenericKernelTy {
115115 }
116116
117117 // / Return maximum block size for maximum occupancy
118- Expected<size_t > maxGroupSize (GenericDeviceTy &Device,
119- size_t DynamicMemSize) const override {
118+ Expected<uint64_t > maxGroupSize (GenericDeviceTy &Device,
119+ uint64_t DynamicMemSize) const override {
120120 // TODO
121121 return 1 ;
122122 }
You can’t perform that action at this time.
0 commit comments