[Fix] Fix CUDA compilation error (#98)

This PR fixes the cuda compilation error.
mlc-ai · Nov 25, 2024 · 590eace · 590eace
1 parent bea9cbc
commit 590eace
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cpp/json_schema_converter.cc b/cpp/json_schema_converter.cc
@@ -667,7 +667,7 @@ std::string JSONSchemaConverter::VisitInteger(
           "exclusiveMaximum",
       }
   );
-  return "(\"0\" | \"-\"? [1-9] [0-9]*) \".0\"?";
+  return "(\"0\" | \"-\"? [1-9] [0-9]*)";
 }
 
 std::string JSONSchemaConverter::VisitNumber(

diff --git a/python/xgrammar/kernels/apply_token_bitmask_inplace_cuda.py b/python/xgrammar/kernels/apply_token_bitmask_inplace_cuda.py
@@ -32,7 +32,7 @@
 
 _apply_token_bitmask_inplace_kernel = """
 #include <cuda_fp16.h>
-
+#include <cuda/std/cstdint>
 #include <cuda/std/limits>
 
 #define BITS_PER_BLOCK 32