Replace all AT_ASSERTM under c10/ (except Exception.h) (pytorch#50843)

Summary:
Pull Request resolved: pytorch#50843

AT_ASSERTM is deprecated and should be replaced by either TORCH_CHECK or
TORCH_INTERNAL_ASSERT, depending on the situation.

Test Plan: Imported from OSS

Reviewed By: ailzhang

Differential Revision: D26074365

Pulled By: ezyang

fbshipit-source-id: 46e13588fad4e24828f3cc99635e9cb2223a6c2c

Author: xuhdev

c10/core/Allocator.cpp

@@ -30,7 +30,7 @@ void SetAllocator(at::DeviceType t, at::Allocator* alloc, uint8_t priority) {
 
 at::Allocator* GetAllocator(const at::DeviceType& t) {
   auto* alloc = allocator_array[static_cast<int>(t)];
-  AT_ASSERTM(alloc, "Allocator for ", t, " is not set.");
+  TORCH_INTERNAL_ASSERT(alloc, "Allocator for ", t, " is not set.");
   return alloc;
 }
 

c10/core/StorageImpl.h

@@ -23,7 +23,7 @@ struct C10_API StorageImpl final : public c10::intrusive_ptr_target {
         received_cuda_(false),
         allocator_(allocator) {
     if (resizable) {
-      AT_ASSERTM(
+      TORCH_INTERNAL_ASSERT(
           allocator_, "For resizable storage, allocator must be provided");
     }
   }

c10/core/TensorOptions.h

@@ -708,7 +708,7 @@ inline DeviceType computeDeviceType(DispatchKey tid) {
   } else if (tid == DispatchKey::QuantizedXPU) {
     return DeviceType::XPU;
   } else {
-    AT_ASSERTM(false, "Unknown DispatchKey: ", tid);
+    TORCH_INTERNAL_ASSERT(false, "Unknown DispatchKey: ", tid);
   }
 }
 

c10/cuda/CUDAStream.cpp

@@ -175,7 +175,7 @@ static StreamId CUDAStream_getStreamId(const LeakyStreamInternals* ptr) {
         StreamIdType::HIGH, ptr - high_priority_streams[device_index].data());
   }
 
-  AT_ASSERTM(
+  TORCH_INTERNAL_ASSERT(
       0,
       "Could not compute stream ID for ",
       ptr,
@@ -197,7 +197,7 @@ static void initGlobalStreamState() {
   num_gpus = device_count();
   // Check if the number of GPUs matches the expected compile-time max number
   // of GPUs.
-  AT_ASSERTM(
+  TORCH_CHECK(
       num_gpus <= C10_COMPILE_TIME_MAX_GPUS,
       "Number of CUDA devices on the machine is larger than the compiled "
       "max number of gpus expected (",
@@ -269,7 +269,7 @@ LeakyStreamInternals* CUDAStream_internals(CUDAStream s) {
   size_t si = streamIdIndex(s.unwrap().id());
   switch (st) {
     case StreamIdType::DEFAULT:
-      AT_ASSERTM(
+      TORCH_INTERNAL_ASSERT(
           si == 0,
           "Unrecognized stream ",
           s.unwrap(),
@@ -284,7 +284,7 @@ LeakyStreamInternals* CUDAStream_internals(CUDAStream s) {
     case StreamIdType::HIGH:
       return &high_priority_streams[device_index][si];
     default:
-      AT_ASSERTM(
+      TORCH_INTERNAL_ASSERT(
           0,
           "Unrecognized stream ",
           s.unwrap(),